Update app.py
Browse files
app.py
CHANGED
@@ -1,3 +1,4 @@
|
|
|
|
1 |
from huggingface_hub import InferenceClient
|
2 |
import gradio as gr
|
3 |
|
@@ -5,14 +6,16 @@ client = InferenceClient(
|
|
5 |
"mistralai/Mistral-7B-Instruct-v0.3"
|
6 |
)
|
7 |
|
|
|
|
|
8 |
|
9 |
def format_prompt(message, history):
|
10 |
-
|
11 |
-
|
12 |
-
|
13 |
-
|
14 |
-
|
15 |
-
|
16 |
|
17 |
def generate(
|
18 |
prompt, history, system_prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,
|
@@ -38,10 +41,14 @@ def generate(
|
|
38 |
for response in stream:
|
39 |
output += response.token.text
|
40 |
yield output
|
41 |
-
return output
|
42 |
|
|
|
|
|
|
|
|
|
|
|
43 |
|
44 |
-
additional_inputs=[
|
45 |
gr.Textbox(
|
46 |
label="System Prompt",
|
47 |
max_lines=1,
|
@@ -85,8 +92,7 @@ additional_inputs=[
|
|
85 |
)
|
86 |
]
|
87 |
|
88 |
-
examples=[
|
89 |
-
]
|
90 |
|
91 |
gr.ChatInterface(
|
92 |
fn=generate,
|
|
|
1 |
+
import pyttsx3
|
2 |
from huggingface_hub import InferenceClient
|
3 |
import gradio as gr
|
4 |
|
|
|
6 |
"mistralai/Mistral-7B-Instruct-v0.3"
|
7 |
)
|
8 |
|
9 |
+
# Инициализация диктора
|
10 |
+
engine = pyttsx3.init()
|
11 |
|
12 |
def format_prompt(message, history):
|
13 |
+
prompt = "Ты — персонаж в текстовой игре, тебя зовут Рассказчик, взаимодействующий с игроком на русском языке. Каждый твой ответ должен продолжать сюжет игры, принимая во внимание прошлые действия игрока и текущую ситуацию. Твой стиль общения должен быть интересным и поддерживать атмосферу игры. Ты должен поддерживать динамичность сюжета и постоянно добавлять элементы интерактивности и новых вызовов. В начале игры игроку предоставляется информация о мире, его текущем положении и доступных действиях. Он может выбирать из предложенных вариантов или описывать свои действия свободно. Важно, чтобы ты адаптировал ответы в зависимости от решений игрока и развивал сюжет на основе их выбора. Каждый ответ должен быть логичным продолжением событий, с описанием последствий выбранных действий."
|
14 |
+
for user_prompt, bot_response in history:
|
15 |
+
prompt += f"[INST] {user_prompt} [/INST]"
|
16 |
+
prompt += f" {bot_response}</s> "
|
17 |
+
prompt += f"[INST] {message} [/INST]"
|
18 |
+
return prompt
|
19 |
|
20 |
def generate(
|
21 |
prompt, history, system_prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,
|
|
|
41 |
for response in stream:
|
42 |
output += response.token.text
|
43 |
yield output
|
|
|
44 |
|
45 |
+
# Озвучивание ответа
|
46 |
+
engine.say(output)
|
47 |
+
engine.runAndWait()
|
48 |
+
|
49 |
+
return output
|
50 |
|
51 |
+
additional_inputs = [
|
52 |
gr.Textbox(
|
53 |
label="System Prompt",
|
54 |
max_lines=1,
|
|
|
92 |
)
|
93 |
]
|
94 |
|
95 |
+
examples = []
|
|
|
96 |
|
97 |
gr.ChatInterface(
|
98 |
fn=generate,
|