LeeKinXUn commited on
Commit
3dd06f5
1 Parent(s): 072ee27

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -59,7 +59,10 @@ document_store.write_documents(docs_to_index)
59
  def haysstack(input,retriever=retriever):
60
  pipe = FAQPipeline(retriever=retriever)
61
  prediction = pipe.run(query=input, params={"Retriever": {"top_k": 1}})
62
- return prediction['answers'][0]
 
 
 
63
  # Run any question and change top_k to see more or less answers
64
 
65
  import gradio as gr
 
59
  def haysstack(input,retriever=retriever):
60
  pipe = FAQPipeline(retriever=retriever)
61
  prediction = pipe.run(query=input, params={"Retriever": {"top_k": 1}})
62
+ answer = str(prediction['answers'][0])
63
+ ans = answer.split(',')[0]
64
+ return ans[19:]
65
+ #prediction['answers'][0]
66
  # Run any question and change top_k to see more or less answers
67
 
68
  import gradio as gr