Update app.py
Browse files
app.py
CHANGED
|
@@ -6,7 +6,7 @@ from transformers import pipeline
|
|
| 6 |
generator = pipeline('text-generation', model='gpt2')
|
| 7 |
|
| 8 |
def chatbot(YourPrompt):
|
| 9 |
-
response = generator(YourPrompt, max_length=
|
| 10 |
return response
|
| 11 |
|
| 12 |
demo = gr.Interface(fn=chatbot, inputs="text", outputs="text")
|
|
|
|
| 6 |
generator = pipeline('text-generation', model='gpt2')
|
| 7 |
|
| 8 |
def chatbot(YourPrompt):
|
| 9 |
+
response = generator(YourPrompt, max_length=500, num_return_sequences=1)[0]['generated_text']
|
| 10 |
return response
|
| 11 |
|
| 12 |
demo = gr.Interface(fn=chatbot, inputs="text", outputs="text")
|