MarcBrun commited on
Commit
772bcf5
1 Parent(s): fc84913

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -2,11 +2,11 @@ import gradio as gr
2
  from transformers import pipeline
3
  import torch
4
 
 
 
5
  def answer_question(question, answer_text):
6
- model_checkpoint = "MarcBrun/ixambert-finetuned-squad"
7
- question_answerer = pipeline("question-answering", model=model_checkpoint)
8
 
9
- answer = question_answerer(question=question, context=answer_text)
10
 
11
  return answer["answer"]
12
 
 
2
  from transformers import pipeline
3
  import torch
4
 
5
+ qa = pipeline("question-answering", model="MarcBrun/ixambert-finetuned-squad")
6
+
7
  def answer_question(question, answer_text):
 
 
8
 
9
+ answer = qa(question=question, context=answer_text)
10
 
11
  return answer["answer"]
12