Spaces:
Build error
Build error
Fix
Browse files
app.py
CHANGED
|
@@ -2,10 +2,10 @@ from transformers import pipeline, set_seed
|
|
| 2 |
from random import randint
|
| 3 |
import gradio as gr
|
| 4 |
|
| 5 |
-
generator = pipeline('text-generation', model='gpt2-
|
| 6 |
-
set_seed(randint(randint(1000,10000),randint(50000,300000)))
|
| 7 |
|
| 8 |
def gpt2(string, max_length):
|
|
|
|
| 9 |
return generator(string, max_length=max_length, num_return_sequences=1)[0]['generated_text']
|
| 10 |
|
| 11 |
max_length_slider = gr.inputs.Slider(minimum=50, maximum=500, step=10, default=100, label="Maximum Length")
|
|
|
|
| 2 |
from random import randint
|
| 3 |
import gradio as gr
|
| 4 |
|
| 5 |
+
generator = pipeline('text-generation', model='gpt2-medium')
|
|
|
|
| 6 |
|
| 7 |
def gpt2(string, max_length):
|
| 8 |
+
set_seed(randint(randint(1000,10000),randint(50000,300000)))
|
| 9 |
return generator(string, max_length=max_length, num_return_sequences=1)[0]['generated_text']
|
| 10 |
|
| 11 |
max_length_slider = gr.inputs.Slider(minimum=50, maximum=500, step=10, default=100, label="Maximum Length")
|