Update app.py
Browse files
app.py
CHANGED
|
@@ -6,7 +6,7 @@ model_name = "EleutherAI/gpt-neo-125M"
|
|
| 6 |
generator = pipeline("text-generation", model=model_name)
|
| 7 |
|
| 8 |
def generate_text(prompt):
|
| 9 |
-
return generator(prompt, max_length=
|
| 10 |
|
| 11 |
interface = gr.Interface(fn=generate_text, inputs="text", outputs="text")
|
| 12 |
interface.launch()
|
|
|
|
| 6 |
generator = pipeline("text-generation", model=model_name)
|
| 7 |
|
| 8 |
def generate_text(prompt):
|
| 9 |
+
return generator(prompt, max_length=200)[0]["generated_text"]
|
| 10 |
|
| 11 |
interface = gr.Interface(fn=generate_text, inputs="text", outputs="text")
|
| 12 |
interface.launch()
|