dgnk007 commited on
Commit
bd1448a
·
1 Parent(s): e9a5f51

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -10,8 +10,8 @@ model_name_or_path = "dgnk007/crow"
10
  def chatbot(message):
11
  prompt_template=f'### instruct: {message} \n### Bot: '
12
  generate=pipeline('text-generation',model=model_name_or_path)
13
- response=generate(prompt_template,max_length=128,return_full_text=False)
14
- return response[0]['generated_text']
15
 
16
- demo = gr.Interface(fn=chatbot, inputs="textbox", outputs="textbox")
17
  demo.launch()
 
10
  def chatbot(message):
11
  prompt_template=f'### instruct: {message} \n### Bot: '
12
  generate=pipeline('text-generation',model=model_name_or_path)
13
+ response=generate(prompt_template,max_length=128,return_full_text=False,num_return_sequences=2)
14
+ return response[0]['generated_text'], response[1]['generated_text']
15
 
16
+ demo = gr.Interface(fn=chatbot, inputs="textbox", outputs=["textbox", "textbox"])
17
  demo.launch()