wop commited on
Commit
dc2f312
1 Parent(s): 45eb2fa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -11
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
- [gr.inputs.Textbox(label="Context",lines=10),gr.inputs.Dropdown(choices=["longformer-base-4096-finetuned-squadv1","longformer-large-4096-finetuned-triviaqa"], type="value", default="longformer-base-4096-finetuned-squadv1", label="model"),gr.inputs.Textbox(label="Question Answering")],
29
- [gr.outputs.Textbox(label="Output")],
 
 
 
 
 
 
 
 
 
30
  examples=examples,
31
  article=article,
32
  title=title,
33
- description=description).launch(enable_queue=True)
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)