bstraehle commited on
Commit
c01aa5d
·
1 Parent(s): bb3c29a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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"
@@ -36,7 +36,7 @@ YOUTUBE_URL = "https://www.youtube.com/watch?v=--khbXchTeE"
36
  MODEL_NAME = "gpt-4"
37
 
38
  def invoke(openai_api_key, use_rag, prompt):
39
- print(os.listdir("../"))
40
  llm = ChatOpenAI(model_name = MODEL_NAME,
41
  openai_api_key = openai_api_key,
42
  temperature = 0)
 
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"
 
36
  MODEL_NAME = "gpt-4"
37
 
38
  def invoke(openai_api_key, use_rag, prompt):
39
+ print(os.listdir("../app/"))
40
  llm = ChatOpenAI(model_name = MODEL_NAME,
41
  openai_api_key = openai_api_key,
42
  temperature = 0)