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,
|