juanpasanper
commited on
Commit
•
6ce726d
1
Parent(s):
9caee9d
Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,8 @@
|
|
1 |
import gradio as gr
|
|
|
|
|
2 |
def question_answer(context, question):
|
3 |
-
predictions, raw_outputs =
|
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"],
|