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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -9
app.py CHANGED
@@ -23,19 +23,23 @@ def predict(message, history):
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"):
 
23
  try:
24
  response = model(full_prompt)
25
 
26
+ # Глубокая проверка структуры (извлекаем текст любой ценой)
27
+ if isinstance(response, list):
28
+ # Если это список, берем первый элемент
29
+ res_item = response[0]
30
+ if isinstance(res_item, dict):
31
+ # Если внутри словаря есть ключ 'generated_text' (стандарт Novita/HuggingFace)
32
+ return res_item.get("generated_text", str(res_item))
33
+ return str(res_item)
34
 
35
+ elif isinstance(response, dict):
36
+ # Если это сразу словарь
37
+ return response.get("generated_text", str(response))
38
+
39
  return str(response)
40
 
41
  except Exception as e:
42
+ return f"Ошибка: {str(e)}. Попробуй подождать 10 секунд."
43
 
44
  with gr.Blocks(theme=custom_theme, css=css) as demo:
45
  with gr.Sidebar(elem_id="side-bar"):