andyjosh commited on
Commit
b494fe3
·
verified ·
1 Parent(s): 320b343

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=150, num_return_sequences=1)[0]['generated_text']
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")