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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -13
app.py CHANGED
@@ -1,17 +1,7 @@
1
  from transformers import pipeline
2
 
3
- generator = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.1")
4
-
5
- chat_history = []
6
 
7
  def chat(message):
8
- global chat_history
9
-
10
- # Генерация текста
11
- response = generator(message, max_new_tokens=100, do_sample=True)[0]['generated_text']
12
-
13
- # Добавление в историю
14
- chat_history.append(["Ты", message])
15
- chat_history.append(["FlareGPT", response])
16
-
17
- return chat_history
 
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]]