Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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()
|