Spaces:
Runtime error
Runtime error
Ahsen Khaliq
commited on
Commit
•
3bfdf10
1
Parent(s):
97be3c5
Update app.py
Browse files
app.py
CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
|
|
2 |
|
3 |
title = "Reformer"
|
4 |
|
5 |
-
description = "Gradio Demo for Reformer
|
6 |
|
7 |
article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2001.04451' target='_blank'>Reformer: The Efficient Transformer</a></p>"
|
8 |
|
@@ -12,18 +12,16 @@ examples = [
|
|
12 |
|
13 |
reformerc = gr.Interface.load("huggingface/google/reformer-crime-and-punishment")
|
14 |
|
15 |
-
reformere = gr.Interface.load("huggingface/google/reformer-enwik8")
|
16 |
|
17 |
def inference(text,model):
|
18 |
if model == "reformer-crime-and-punishment":
|
19 |
outtext = reformerc(text)
|
20 |
-
else:
|
21 |
-
outtext = reformere(text)
|
22 |
return outtext
|
23 |
|
24 |
gr.Interface(
|
25 |
inference,
|
26 |
-
[gr.inputs.Textbox(label="Input"),gr.inputs.Dropdown(choices=["reformer-crime-and-punishment","reformer-enwik8"], type="value", default="reformer-
|
27 |
gr.outputs.Textbox(label="Output"),
|
28 |
examples=examples,
|
29 |
article=article,
|
|
|
2 |
|
3 |
title = "Reformer"
|
4 |
|
5 |
+
description = "Gradio Demo for Reformer. To use it, simply add your text, or click one of the examples to load them. Read more at the links below."
|
6 |
|
7 |
article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2001.04451' target='_blank'>Reformer: The Efficient Transformer</a></p>"
|
8 |
|
|
|
12 |
|
13 |
reformerc = gr.Interface.load("huggingface/google/reformer-crime-and-punishment")
|
14 |
|
15 |
+
#reformere = gr.Interface.load("huggingface/google/reformer-enwik8")
|
16 |
|
17 |
def inference(text,model):
|
18 |
if model == "reformer-crime-and-punishment":
|
19 |
outtext = reformerc(text)
|
|
|
|
|
20 |
return outtext
|
21 |
|
22 |
gr.Interface(
|
23 |
inference,
|
24 |
+
[gr.inputs.Textbox(label="Input"),gr.inputs.Dropdown(choices=["reformer-crime-and-punishment","reformer-enwik8"], type="value", default="reformer-crime-and-punishment", label="model")],
|
25 |
gr.outputs.Textbox(label="Output"),
|
26 |
examples=examples,
|
27 |
article=article,
|