Spaces:
Sleeping
Sleeping
app.py
CHANGED
@@ -8,7 +8,7 @@ generator = pipeline('text-generation', model='gpt2', return_full_text=False, do
|
|
8 |
|
9 |
|
10 |
def Bemenet(input_string):
|
11 |
-
return generator(os.environ['SECRET_PROMPT'] + " " + input_string, max_length=100
|
12 |
|
13 |
textbox = gr.Textbox()
|
14 |
|
|
|
8 |
|
9 |
|
10 |
def Bemenet(input_string):
|
11 |
+
return generator(os.environ['SECRET_PROMPT'] + " " + input_string, max_length=100)[0]['generated_text']
|
12 |
|
13 |
textbox = gr.Textbox()
|
14 |
|