bstraehle commited on
Commit
d6396cd
·
1 Parent(s): 95a4e15

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -1,5 +1,5 @@
1
  import gradio as gr
2
- import os, time
3
 
4
  from dotenv import load_dotenv, find_dotenv
5
 
@@ -11,15 +11,18 @@ _ = load_dotenv(find_dotenv())
11
 
12
  RAG_INGESTION = False # load, split, embed, and store documents
13
 
 
 
 
 
14
  config = {
15
  "k": 3, # retrieve documents
16
  "model_name": "gpt-4-0314", # llm
17
  "temperature": 0 # llm
18
  }
19
 
20
- RAG_OFF = "Off"
21
- RAG_LANGCHAIN = "LangChain"
22
- RAG_LLAMAINDEX = "LlamaIndex"
23
 
24
  def invoke(openai_api_key, prompt, rag_option):
25
  if (openai_api_key == ""):
 
1
  import gradio as gr
2
+ import logging, os, sys, time
3
 
4
  from dotenv import load_dotenv, find_dotenv
5
 
 
11
 
12
  RAG_INGESTION = False # load, split, embed, and store documents
13
 
14
+ RAG_OFF = "Off"
15
+ RAG_LANGCHAIN = "LangChain"
16
+ RAG_LLAMAINDEX = "LlamaIndex"
17
+
18
  config = {
19
  "k": 3, # retrieve documents
20
  "model_name": "gpt-4-0314", # llm
21
  "temperature": 0 # llm
22
  }
23
 
24
+ logging.basicConfig(stream = sys.stdout, level = logging.INFO)
25
+ logging.getLogger().addHandler(logging.StreamHandler(stream = sys.stdout))
 
26
 
27
  def invoke(openai_api_key, prompt, rag_option):
28
  if (openai_api_key == ""):