Neu256 commited on
Commit
845959c
·
verified ·
1 Parent(s): 91a6273

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ def model_generate(text):
15
  ]
16
  prompt = pipe.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
17
  outputs = pipe(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
18
- return outputs
19
 
20
  iface = gr.Interface(fn=model_generate, inputs=["text"], outputs="text")
21
  iface.launch()
 
15
  ]
16
  prompt = pipe.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
17
  outputs = pipe(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
18
+ return outputs[0]["generated_text"]
19
 
20
  iface = gr.Interface(fn=model_generate, inputs=["text"], outputs="text")
21
  iface.launch()