Update app.py

app.py

@@ -1,23 +1,46 @@
 import gradio as gr
 from transformers import pipeline
+import sys
 
-print("Loading model...")
-pipe = pipeline("text-generation", model="Andrewstivan/AURA", device=-1)
-print("Model loaded!")
+print("Loading model...", flush=True)
+try:
+    pipe = pipeline(
+        "text-generation", 
+        model="Andrewstivan/AURA", 
+        device=-1,
+        trust_remote_code=True 
+    )
+    print("Model loaded successfully!", flush=True)
+except Exception as e:
+    print(f"Error loading model: {e}", flush=True)
+    sys.exit(1)
 
 def chat(message, history):
-    result = pipe(
-        message,
-        max_new_tokens=300,
-        temperature=0.8,
-        do_sample=True,
-        truncation=True,
-        max_length=512
-    )
-    return result[0]["generated_text"]
+    if not message:
+        return ""
+    
+    try:
+        result = pipe(
+            message,
+            max_new_tokens=300,
+            temperature=0.8,
+            do_sample=True,
+            truncation=True,
+            max_length=512
+        )
+        return result[0]["generated_text"]
+    except Exception as e:
+        return f"Ошибка: {e}"
 
-gr.ChatInterface(
+
+demo = gr.ChatInterface(
     fn=chat,
     title="✨ Aura — поэтичная модель",
     description="Пишите по-русски, модель отвечает на русском"
-).launch()
+)
+
+demo.launch(
+    server_name="0.0.0.0",  
+    server_port=7860,        
+    share=False              
+)