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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,5 +1,5 @@
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",}],}])
 
1
  import gradio as gr
2
+ model = Model.from_pretrained(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",}],}])