Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -4,11 +4,11 @@ from transformers import pipeline
|
|
4 |
generator = pipeline('text-generation', model='gpt2')
|
5 |
|
6 |
def func(sentence, max_length, temp):
|
7 |
-
output_list = generator(sentence, max_length=max_length, num_return_sequences=5, temperature=temp)
|
8 |
output_strs = [dict['generated_text'] for dict in output_list]
|
9 |
return output_strs
|
10 |
|
11 |
-
demo = gr.Interface(fn=func, inputs=["text", gr.Slider(5,
|
12 |
|
13 |
if __name__ == "__main__":
|
14 |
demo.launch()
|
|
|
4 |
generator = pipeline('text-generation', model='gpt2')
|
5 |
|
6 |
def func(sentence, max_length, temp):
|
7 |
+
output_list = generator(sentence, max_length=max_length, num_return_sequences=5, temperature=float(temp))
|
8 |
output_strs = [dict['generated_text'] for dict in output_list]
|
9 |
return output_strs
|
10 |
|
11 |
+
demo = gr.Interface(fn=func, inputs=["text", gr.Slider(5, 25), gr.Slider(0.1, 100)], outputs=["text", "text", "text", "text", "text"])
|
12 |
|
13 |
if __name__ == "__main__":
|
14 |
demo.launch()
|