eaglelandsonce commited on
Commit
9bc5b30
·
verified ·
1 Parent(s): a5221b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -39,11 +39,13 @@ else:
39
 
40
 
41
  # Setup Clarifai Vector DB
42
- clarifai_vector_db = Clarifai(
43
  user_id=USER_ID,
44
  app_id=APP_ID,
45
- number_of_docs=NUMBER_OF_DOCS,
46
- )
 
 
47
 
48
  # Setup Clarifai LLM
49
  USER_ID = "openai"
@@ -51,6 +53,8 @@ APP_ID = "chat-completion"
51
  MODEL_ID = "GPT-3_5-turbo"
52
  clarifai_llm = Clarifai(pat=CLARIFAI_PAT, user_id=USER_ID, app_id=APP_ID, model_id=MODEL_ID)
53
 
 
 
54
  # Initialize RetrievalQA
55
  qa = RetrievalQA.from_chain_type(
56
  llm=clarifai_llm,
 
39
 
40
 
41
  # Setup Clarifai Vector DB
42
+ clarifai_vector_db = Clarifai.from_documents(
43
  user_id=USER_ID,
44
  app_id=APP_ID,
45
+ documents=docs,
46
+ pat=CLARIFAI_PAT,
47
+ number_of_docs=3,
48
+ )
49
 
50
  # Setup Clarifai LLM
51
  USER_ID = "openai"
 
53
  MODEL_ID = "GPT-3_5-turbo"
54
  clarifai_llm = Clarifai(pat=CLARIFAI_PAT, user_id=USER_ID, app_id=APP_ID, model_id=MODEL_ID)
55
 
56
+
57
+
58
  # Initialize RetrievalQA
59
  qa = RetrievalQA.from_chain_type(
60
  llm=clarifai_llm,