Update app.py

app.py

@@ -2,10 +2,20 @@ import gradio as gr
 from transformers import pipeline
 
 # Завантаження моделі
-generator = pipeline("text-generation", model="EleutherAI/gpt-neo-1.3B")
+def load_model():
+    try:
+        generator = pipeline("text-generation", model="EleutherAI/gpt-neo-1.3B")
+        return generator
+    except Exception as e:
+        return f"Помилка завантаження моделі: {e}"
+
+# Ініціалізація моделі
+generator = load_model()
 
 # Функція для генерації відповіді
 def chat_with_npc(user_input):
+    if isinstance(generator, str):  # Якщо модель не завантажилась
+        return generator
     response = generator(user_input, max_length=100, do_sample=True, temperature=0.7)
     return response[0]["generated_text"]
 
@@ -20,4 +30,4 @@ interface = gr.Interface(
 
 # Запуск програми
 if __name__ == "__main__":
-    interface.launch()
+    interface.launch()