HaveAI commited on
Commit
34082b9
·
verified ·
1 Parent(s): 809c7df

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -12
app.py CHANGED
@@ -1,7 +1,6 @@
1
  import gradio as gr
2
 
3
- # Загружаем модель один раз
4
- # Это самый стабильный способ для бесплатных Space
5
  model = gr.load("models/moonshotai/Kimi-K2-Thinking", provider="novita")
6
 
7
  custom_theme = gr.themes.Soft(
@@ -18,24 +17,25 @@ css = """
18
  """
19
 
20
  def predict(message, history):
21
- # Системный промпт
22
  system_instr = "Тебя зовут Gemini. "
23
  full_prompt = system_instr + message
24
 
25
  try:
26
- # Вызываем модель
27
  response = model(full_prompt)
28
 
29
- # Проверка: если пришел список, берем первый элемент
30
- if isinstance(response, list):
31
- return response[0]
32
- # Если пришел словарь (бывает в некоторых API)
33
- elif isinstance(response, dict):
34
- return response.get("generated_text", str(response))
 
 
 
 
35
 
36
- return response
37
  except Exception as e:
38
- return f"Ошибка: {str(e)}. Попробуй подождать 10 секунд и отправить снова."
39
 
40
  with gr.Blocks(theme=custom_theme, css=css) as demo:
41
  with gr.Sidebar(elem_id="side-bar"):
 
1
  import gradio as gr
2
 
3
+ # Загружаем модель
 
4
  model = gr.load("models/moonshotai/Kimi-K2-Thinking", provider="novita")
5
 
6
  custom_theme = gr.themes.Soft(
 
17
  """
18
 
19
  def predict(message, history):
 
20
  system_instr = "Тебя зовут Gemini. "
21
  full_prompt = system_instr + message
22
 
23
  try:
 
24
  response = model(full_prompt)
25
 
26
+ # РЕШЕНИЕ ОШИБКИ ТУТ:
27
+ # Провайдеры часто возвращают список словарей: [{'generated_text': 'ответ'}]
28
+ if isinstance(response, list) and len(response) > 0:
29
+ item = response[0]
30
+ if isinstance(item, dict) and "generated_text" in item:
31
+ return item["generated_text"]
32
+ else:
33
+ return str(item) # На случай, если там просто строка
34
+
35
+ return str(response)
36
 
 
37
  except Exception as e:
38
+ return f"Ошибка: {str(e)}. Попробуй еще раз через 10 секунд."
39
 
40
  with gr.Blocks(theme=custom_theme, css=css) as demo:
41
  with gr.Sidebar(elem_id="side-bar"):