HelloSun commited on
Commit
bb0ec8e
·
verified ·
1 Parent(s): 3188082

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -13,10 +13,10 @@ pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
13
 
14
  def respond(message, history):
15
  # 將當前訊息與歷史訊息合併
16
- input_text = message if not history else history[-1]["content"] + " " + message
17
-
18
  # 獲取模型的回應
19
- response = pipe(input_text, max_length=100, truncation=True, num_return_sequences=1)
20
  reply = response[0]['generated_text']
21
 
22
  # 返回新的消息格式
 
13
 
14
  def respond(message, history):
15
  # 將當前訊息與歷史訊息合併
16
+ #input_text = message if not history else history[-1]["content"] + " " + message
17
+ input_text = message
18
  # 獲取模型的回應
19
+ response = pipe(input_text, max_length=500, truncation=True, num_return_sequences=1)
20
  reply = response[0]['generated_text']
21
 
22
  # 返回新的消息格式