Update app.py

app.py

@@ -1,5 +1,9 @@
 import gradio as gr
 from transformers import pipeline
+import os
+os.environ["HF_HOME"] = "/data/.huggingface"
+os.environ["HF_DATASETS_CACHE"] = "/data/.cache/huggingface/datasets"
+os.environ["TRANSFORMERS_CACHE"] = "/data/.cache/huggingface/transformers"
 
 print("Loading model...")
 pipe = pipeline("text-generation", model="Andrewstivan/AURA", device=-1)
@@ -20,4 +24,6 @@ gr.ChatInterface(
     fn=chat,
     title="✨ Aura — поэтичная модель",
     description="Пишите по-русски, модель отвечает на русском"
-).launch()
+).launch()
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0", server_port=7860)