Spaces:
Sleeping
Sleeping
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")
|