Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -15,22 +15,29 @@ io1 = gr.Interface.load("huggingface/valhalla/longformer-base-4096-finetuned-squ
|
|
15 |
io2 = gr.Interface.load("huggingface/allenai/longformer-large-4096-finetuned-triviaqa")
|
16 |
|
17 |
|
18 |
-
def inference(context, model,question):
|
19 |
if model == "longformer-base-4096-finetuned-squadv1":
|
20 |
-
outlabel = io1(context,question)
|
21 |
else:
|
22 |
-
outlabel = io2(context,question)
|
23 |
return outlabel
|
24 |
-
|
25 |
|
26 |
gr.Interface(
|
27 |
-
inference,
|
28 |
-
[
|
29 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
30 |
examples=examples,
|
31 |
article=article,
|
32 |
title=title,
|
33 |
-
description=description
|
34 |
-
|
35 |
-
|
36 |
-
|
|
|
15 |
io2 = gr.Interface.load("huggingface/allenai/longformer-large-4096-finetuned-triviaqa")
|
16 |
|
17 |
|
18 |
+
def inference(context, model, question):
|
19 |
if model == "longformer-base-4096-finetuned-squadv1":
|
20 |
+
outlabel = io1(context, question)
|
21 |
else:
|
22 |
+
outlabel = io2(context, question)
|
23 |
return outlabel
|
24 |
+
|
25 |
|
26 |
gr.Interface(
|
27 |
+
inference,
|
28 |
+
[
|
29 |
+
gr.Textbox(label="Context", lines=10),
|
30 |
+
gr.Dropdown(
|
31 |
+
choices=["longformer-base-4096-finetuned-squadv1", "longformer-large-4096-finetuned-triviaqa"],
|
32 |
+
type="value",
|
33 |
+
default="longformer-base-4096-finetuned-squadv1",
|
34 |
+
label="model",
|
35 |
+
),
|
36 |
+
gr.Textbox(label="Question Answering"),
|
37 |
+
],
|
38 |
+
[gr.Textbox(label="Output")],
|
39 |
examples=examples,
|
40 |
article=article,
|
41 |
title=title,
|
42 |
+
description=description,
|
43 |
+
).launch(enable_queue=True)
|
|
|
|