Spaces:
Runtime error
Runtime error
Commit
·
e9e8d18
1
Parent(s):
d771d63
Update app.py
Browse files
app.py
CHANGED
@@ -2,13 +2,12 @@ import gradio as gr
|
|
2 |
from transformers import pipeline
|
3 |
|
4 |
|
5 |
-
|
6 |
def generator_gpt2(text):
|
7 |
gen = pipeline('text-generation', model='gpt2')
|
8 |
-
return gen(text, max_length=
|
9 |
def generator_openai_gpt(text):
|
10 |
gen = pipeline('text-generation', model='openai-gpt')
|
11 |
-
return gen(text, max_length=
|
12 |
|
13 |
examples=[["Hello, I'm a language model,"],
|
14 |
["How are you?\n"]]
|
|
|
2 |
from transformers import pipeline
|
3 |
|
4 |
|
|
|
5 |
def generator_gpt2(text):
|
6 |
gen = pipeline('text-generation', model='gpt2')
|
7 |
+
return gen(text, max_length=30, num_return_sequences=1)[0]['generated_text']
|
8 |
def generator_openai_gpt(text):
|
9 |
gen = pipeline('text-generation', model='openai-gpt')
|
10 |
+
return gen(text, max_length=30, num_return_sequences=1)[0]['generated_text']
|
11 |
|
12 |
examples=[["Hello, I'm a language model,"],
|
13 |
["How are you?\n"]]
|