dgnk007 commited on
Commit
41b6d08
·
1 Parent(s): 84ba798

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,9 +7,9 @@ from transformers import pipeline
7
  model_name_or_path = "dgnk007/crow"
8
 
9
  def generate_response(message):
10
- prompt_template=f'### instruct: {message} \n### Bot: '
11
  generate=pipeline('text-generation',model=model_name_or_path)
12
- response=generate(prompt_template,max_length=128,return_full_text=False,num_return_sequences=2)
13
  return response[0]['generated_text'], response[1]['generated_text']
14
 
15
  with gr.Blocks(theme=gr.themes.Soft()) as block:
 
7
  model_name_or_path = "dgnk007/crow"
8
 
9
  def generate_response(message):
10
+ prompt_template=f"### instruct: {message} \n ### Bot:"
11
  generate=pipeline('text-generation',model=model_name_or_path)
12
+ response=generate(prompt_template,max_length=1024,return_full_text=False,num_return_sequences=2)
13
  return response[0]['generated_text'], response[1]['generated_text']
14
 
15
  with gr.Blocks(theme=gr.themes.Soft()) as block: