bstraehle commited on
Commit
3d46f59
·
1 Parent(s): 006862c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -27,8 +27,8 @@ LLM_CHAIN_PROMPT = PromptTemplate(input_variables = ["question"],
27
  RAG_CHAIN_PROMPT = PromptTemplate(input_variables = ["context", "question"],
28
  template = rag_template)
29
 
30
- CHROMA_DIR = "/data/chroma"
31
- YOUTUBE_DIR = "/data/youtube"
32
 
33
  YOUTUBE_URL = "https://www.youtube.com/watch?v=--khbXchTeE"
34
  #YOUTUBE_URL = "https://www.youtube.com/watch?v=RfvL_423a-I&list=PL2yQDdvlhXf_hIzmfHCdbcXj2hS52oP9r&index=2"
@@ -38,6 +38,7 @@ MODEL_NAME = "gpt-4"
38
  def invoke(openai_api_key, use_rag, prompt):
39
  print(os.listdir("../"))
40
  print(os.listdir("../app/"))
 
41
  llm = ChatOpenAI(model_name = MODEL_NAME,
42
  openai_api_key = openai_api_key,
43
  temperature = 0)
 
27
  RAG_CHAIN_PROMPT = PromptTemplate(input_variables = ["context", "question"],
28
  template = rag_template)
29
 
30
+ CHROMA_DIR = "/data"
31
+ YOUTUBE_DIR = "/data"
32
 
33
  YOUTUBE_URL = "https://www.youtube.com/watch?v=--khbXchTeE"
34
  #YOUTUBE_URL = "https://www.youtube.com/watch?v=RfvL_423a-I&list=PL2yQDdvlhXf_hIzmfHCdbcXj2hS52oP9r&index=2"
 
38
  def invoke(openai_api_key, use_rag, prompt):
39
  print(os.listdir("../"))
40
  print(os.listdir("../app/"))
41
+ print(os.listdir("../data/"))
42
  llm = ChatOpenAI(model_name = MODEL_NAME,
43
  openai_api_key = openai_api_key,
44
  temperature = 0)