Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -2,7 +2,6 @@ import gradio as gr
|
|
2 |
#from transformers import
|
3 |
import tensorflow as tf
|
4 |
from transformers import TFGPT2LMHeadModel, GPT2Tokenizer
|
5 |
-
import math
|
6 |
|
7 |
#generator = pipeline('text-generation', model='gpt2')
|
8 |
|
@@ -19,7 +18,7 @@ def func(sentence, max_length, temperature):
|
|
19 |
output_list = model.generate(
|
20 |
input_ids,
|
21 |
do_sample=True,
|
22 |
-
max_length=
|
23 |
temperature=temperature,
|
24 |
top_p=0.92,
|
25 |
top_k=0
|
@@ -29,7 +28,7 @@ def func(sentence, max_length, temperature):
|
|
29 |
return output_strs
|
30 |
|
31 |
|
32 |
-
demo = gr.Interface(fn=func, inputs=["text", gr.Slider(5, 25), gr.Slider(0.1, 100)], outputs=["text", "text", "text", "text", "text"])
|
33 |
|
34 |
if __name__ == "__main__":
|
35 |
demo.launch()
|
|
|
2 |
#from transformers import
|
3 |
import tensorflow as tf
|
4 |
from transformers import TFGPT2LMHeadModel, GPT2Tokenizer
|
|
|
5 |
|
6 |
#generator = pipeline('text-generation', model='gpt2')
|
7 |
|
|
|
18 |
output_list = model.generate(
|
19 |
input_ids,
|
20 |
do_sample=True,
|
21 |
+
max_length=max_length,
|
22 |
temperature=temperature,
|
23 |
top_p=0.92,
|
24 |
top_k=0
|
|
|
28 |
return output_strs
|
29 |
|
30 |
|
31 |
+
demo = gr.Interface(fn=func, inputs=["text", gr.Slider(5, 25, value=10, step=1), gr.Slider(0.1, 100)], outputs=["text", "text", "text", "text", "text"])
|
32 |
|
33 |
if __name__ == "__main__":
|
34 |
demo.launch()
|