ErikH commited on
Commit
a3e5f1b
1 Parent(s): c2c1450

Update pages/bot.py

Browse files
Files changed (1) hide show
  1. pages/bot.py +1 -1
pages/bot.py CHANGED
@@ -103,7 +103,7 @@ def main():
103
  qa_pipeline = pipeline("question-answering", model="deutsche-telekom/bert-multi-english-german-squad2", tokenizer="deutsche-telekom/bert-multi-english-german-squad2")
104
 
105
  # Frage beantworten
106
- answer = qa_pipeline(question=question, context=context,)
107
 
108
  # Gib die Antwort aus
109
  st.text("Basisantwort:")
 
103
  qa_pipeline = pipeline("question-answering", model="deutsche-telekom/bert-multi-english-german-squad2", tokenizer="deutsche-telekom/bert-multi-english-german-squad2")
104
 
105
  # Frage beantworten
106
+ answer = qa_pipeline(question=question, context=context, top_k=3)
107
 
108
  # Gib die Antwort aus
109
  st.text("Basisantwort:")