chatbytes commited on
Commit
9b707b8
·
verified ·
1 Parent(s): 881d45d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -5
app.py CHANGED
@@ -35,12 +35,16 @@ def text_extract(file):
35
  db = FAISS.from_texts(text_splitter, embeddings);
36
  retriever = db.as_retriever(search_type="similarity", search_kwargs={"k": 2})
37
  llm=GooglePalm(google_api_key=google_api)
38
- qa = RetrievalQA.from_chain_type(
39
- llm=llm, chain_type="stuff", retriever=retriever, return_source_documents=True
40
- )
41
- result=qa.invoke("hi")
 
 
 
 
42
 
43
- return "hi"
44
 
45
 
46
  with gr.Blocks() as demo:
 
35
  db = FAISS.from_texts(text_splitter, embeddings);
36
  retriever = db.as_retriever(search_type="similarity", search_kwargs={"k": 2})
37
  llm=GooglePalm(google_api_key=google_api)
38
+ try:
39
+ qa = RetrievalQA.from_chain_type(
40
+ llm=llm, chain_type="stuff", retriever=retriever, return_source_documents=True
41
+ )
42
+ result=qa.invoke("hi")
43
+ except:
44
+ result="hi"
45
+
46
 
47
+ return result['result']
48
 
49
 
50
  with gr.Blocks() as demo: