Update app.py
Browse files
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 |
-
|
| 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
|
| 147 |
else:
|
| 148 |
-
translated = translate_text(response_text +"; Ms
|
| 149 |
-
|
| 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})
|