mrbui1990 commited on
Commit
d43b399
·
verified ·
1 Parent(s): 99762e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -101,8 +101,9 @@ def chat_with_model(prompt, system_prompt, chatbot_display, internal_history,lan
101
  else:
102
  if lang != None:
103
  prompt = translate_text(prompt,"English")
104
- print("prompt : "+prompt)
105
  prompt = prompt +" [Detailed description of the physical actions and expressions.]"
 
106
  # 1. Khởi tạo nếu đây là lần chạy đầu tiên
107
  # chatbot_display là [[user_msg, ai_msg], ...]
108
  if chatbot_display is None:
@@ -140,13 +141,14 @@ def chat_with_model(prompt, system_prompt, chatbot_display, internal_history,lan
140
 
141
  # 5. Decode *chỉ* phần trả lời mới
142
  response_text = tokenizer.decode(output_tokens[0][inputs.shape[-1]:], skip_special_tokens=True)
 
143
  translated = response_text
144
  if lang != None :
145
  if gender :
146
- translated = translate_text(response_text+"; Mr Tuan",lang)
147
  else:
148
- translated = translate_text(response_text +"; Ms Jenny",lang)
149
- print("response_text : "+response_text)
150
  print("translated : "+translated)
151
  # 6. Cập nhật "bộ nhớ" (gr.State) với lượt nói MỚI
152
  internal_history.append({"role": "user", "content": prompt})
 
101
  else:
102
  if lang != None:
103
  prompt = translate_text(prompt,"English")
104
+
105
  prompt = prompt +" [Detailed description of the physical actions and expressions.]"
106
+ print("prompt : "+prompt)
107
  # 1. Khởi tạo nếu đây là lần chạy đầu tiên
108
  # chatbot_display là [[user_msg, ai_msg], ...]
109
  if chatbot_display is None:
 
141
 
142
  # 5. Decode *chỉ* phần trả lời mới
143
  response_text = tokenizer.decode(output_tokens[0][inputs.shape[-1]:], skip_special_tokens=True)
144
+ print("response_text : "+response_text)
145
  translated = response_text
146
  if lang != None :
147
  if gender :
148
+ translated = translate_text(response_text+"; Mr.",lang)
149
  else:
150
+ translated = translate_text(response_text +"; Ms.",lang)
151
+
152
  print("translated : "+translated)
153
  # 6. Cập nhật "bộ nhớ" (gr.State) với lượt nói MỚI
154
  internal_history.append({"role": "user", "content": prompt})