juanpasanper commited on
Commit
6ce726d
1 Parent(s): 9caee9d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,6 +1,8 @@
1
  import gradio as gr
 
 
2
  def question_answer(context, question):
3
- predictions, raw_outputs = este_si_me_sirvio.predict([{"context": context, "qas": [{"question": question, "id": "0",}],}])
4
  prediccion = predictions[0]['answer'][0]
5
  return prediccion
6
  iface = gr.Interface(fn=question_answer, inputs=["text", "text"], outputs=["text"],
 
1
  import gradio as gr
2
+ model = Model(model_name=este_si_me_sirvio.bin)
3
+ model.load_state_dict(torch.load(juanpasanper/tigo_question_answer))
4
  def question_answer(context, question):
5
+ predictions, raw_outputs = model.predict([{"context": context, "qas": [{"question": question, "id": "0",}],}])
6
  prediccion = predictions[0]['answer'][0]
7
  return prediccion
8
  iface = gr.Interface(fn=question_answer, inputs=["text", "text"], outputs=["text"],