HaveAI commited on
Commit
cf407c7
·
verified ·
1 Parent(s): df42d7b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  from transformers import pipeline
2
 
3
- generator = pipeline("text-generation", model="tiiuae/falcon-7b-instruct")
4
 
5
  def chat(message):
6
- result = generator(message, max_new_tokens=100, do_sample=True)[0]["generated_text"]
7
  return [["Ты", message], ["FlareGPT", result]]
 
1
  from transformers import pipeline
2
 
3
+ generator = pipeline("text2text-generation", model="google/flan-t5-small")
4
 
5
  def chat(message):
6
+ result = generator(message, max_new_tokens=100)[0]["generated_text"]
7
  return [["Ты", message], ["FlareGPT", result]]