Xolkin commited on
Commit
42a2d58
·
verified ·
1 Parent(s): b87d619

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -3
app.py CHANGED
@@ -38,8 +38,10 @@ def respond(
38
  response += token
39
  yield response
40
 
41
- # Выделение диагноза жирным шрифтом
42
  diagnosis_start = response.find("Предварительный диагноз:")
 
 
43
  if diagnosis_start != -1:
44
  diagnosis_end = response.find("\n", diagnosis_start)
45
  if diagnosis_end == -1:
@@ -47,6 +49,13 @@ def respond(
47
  diagnosis = response[diagnosis_start:diagnosis_end]
48
  response = response[:diagnosis_start] + f"<b>{diagnosis}</b>" + response[diagnosis_end:]
49
 
 
 
 
 
 
 
 
50
  # Добавление идентификационного сообщения к ответу
51
  final_response = f"{response}\n\nСоздано больницей EMS штата Alta"
52
  yield final_response
@@ -56,10 +65,10 @@ demo = gr.ChatInterface(
56
  respond,
57
  additional_inputs=[
58
  gr.Textbox(
59
- value="Привет! Я помощник врача в больнице EMS штата Alta! Опиши свои симптомы кратко, и я поставлю предварительный диагноз.",
60
  label="Системное сообщение"
61
  ),
62
- gr.Slider(minimum=1, maximum=2048, value=100, step=1, label="Максимальное количество новых токенов"),
63
  gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Температура"),
64
  gr.Slider(
65
  minimum=0.1,
 
38
  response += token
39
  yield response
40
 
41
+ # Выделение диагноза и операции жирным шрифтом
42
  diagnosis_start = response.find("Предварительный диагноз:")
43
+ operation_start = response.find("Рекомендуемая операция:")
44
+
45
  if diagnosis_start != -1:
46
  diagnosis_end = response.find("\n", diagnosis_start)
47
  if diagnosis_end == -1:
 
49
  diagnosis = response[diagnosis_start:diagnosis_end]
50
  response = response[:diagnosis_start] + f"<b>{diagnosis}</b>" + response[diagnosis_end:]
51
 
52
+ if operation_start != -1:
53
+ operation_end = response.find("\n", operation_start)
54
+ if operation_end == -1:
55
+ operation_end = len(response)
56
+ operation = response[operation_start:operation_end]
57
+ response = response[:operation_start] + f"<b>{operation}</b>" + response[operation_end:]
58
+
59
  # Добавление идентификационного сообщения к ответу
60
  final_response = f"{response}\n\nСоздано больницей EMS штата Alta"
61
  yield final_response
 
65
  respond,
66
  additional_inputs=[
67
  gr.Textbox(
68
+ value="Привет! Я помощник врача в больнице EMS штата Alta! Опиши свои симптомы кратко, и я поставлю предварительный диагноз и предложу операцию, если это необходимо.",
69
  label="Системное сообщение"
70
  ),
71
+ gr.Slider(minimum=1, maximum=4096, value=512, step=1, label="Максимальное количество новых токенов"),
72
  gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Температура"),
73
  gr.Slider(
74
  minimum=0.1,