Spaces:
Sleeping
Sleeping
lazarus
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -12,7 +12,30 @@ def load_model():
|
|
12 |
|
13 |
# Ініціалізація моделі
|
14 |
generator = load_model()
|
|
|
15 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
16 |
# Функція для генерації відповіді
|
17 |
def chat_with_npc(user_input):
|
18 |
if isinstance(generator, str): # Якщо модель не завантажилась
|
@@ -36,7 +59,7 @@ interface = gr.Interface(
|
|
36 |
fn=chat_with_npc,
|
37 |
inputs="text",
|
38 |
outputs="text",
|
39 |
-
title="
|
40 |
description="Оптимізований віртуальний помічник для запитів з великою кількістю слів.",
|
41 |
examples=[
|
42 |
["Як можна розвивати духовний потенціал у сучасному світі?"],
|
@@ -46,4 +69,5 @@ interface = gr.Interface(
|
|
46 |
|
47 |
# Запуск програми
|
48 |
if __name__ == "__main__":
|
49 |
-
interface.launch()
|
|
|
|
12 |
|
13 |
# Ініціалізація моделі
|
14 |
generator = load_model()
|
15 |
+
conversation_history = []
|
16 |
|
17 |
+
def chat_with_npc(user_input):
|
18 |
+
global conversation_history
|
19 |
+
if isinstance(generator, str):
|
20 |
+
return generator
|
21 |
+
try:
|
22 |
+
# Додаємо новий запит до історії
|
23 |
+
conversation_history.append({"role": "user", "content": user_input})
|
24 |
+
|
25 |
+
# Формуємо контекст
|
26 |
+
context = "\n".join([f"{msg['role']}: {msg['content']}" for msg in conversation_history])
|
27 |
+
|
28 |
+
# Генеруємо відповідь
|
29 |
+
response = generator(context, max_length=300, num_return_sequences=1, do_sample=True, temperature=0.7, top_p=0.9)
|
30 |
+
bot_reply = response[0]["generated_text"]
|
31 |
+
|
32 |
+
# Додаємо відповідь до історії
|
33 |
+
conversation_history.append({"role": "assistant", "content": bot_reply})
|
34 |
+
return bot_reply
|
35 |
+
except Exception as e:
|
36 |
+
return f"Помилка генерації відповіді: {e}"
|
37 |
+
|
38 |
+
|
39 |
# Функція для генерації відповіді
|
40 |
def chat_with_npc(user_input):
|
41 |
if isinstance(generator, str): # Якщо модель не завантажилась
|
|
|
59 |
fn=chat_with_npc,
|
60 |
inputs="text",
|
61 |
outputs="text",
|
62 |
+
title="Alt Помічник",
|
63 |
description="Оптимізований віртуальний помічник для запитів з великою кількістю слів.",
|
64 |
examples=[
|
65 |
["Як можна розвивати духовний потенціал у сучасному світі?"],
|
|
|
69 |
|
70 |
# Запуск програми
|
71 |
if __name__ == "__main__":
|
72 |
+
interface.launch()
|
73 |
+
|