Update app.py

app.py

@@ -1,16 +1,24 @@
 import subprocess
 import sys
 import threading
+from http.server import HTTPServer, BaseHTTPRequestHandler
 
 def run_training():
     print("Starting training job...")
     subprocess.run([sys.executable, "training.py"], check=True)
     print("Training finished.")
 
-thread = threading.Thread(target=run_training)
-thread.start()
+# Start training in background
+threading.Thread(target=run_training).start()
 
-import gradio as gr
-def status():
-    return "Training running…"
-gr.Interface(fn=status, inputs=[], outputs="text").launch()
+# Minimal HTTP server to keep the container alive (port 7860)
+class Handler(BaseHTTPRequestHandler):
+    def do_GET(self):
+        self.send_response(200)
+        self.end_headers()
+        self.wfile.write(b"Training in progress...")
+    def do_POST(self):
+        self.do_GET()
+
+print("Keeping container alive on port 7860")
+HTTPServer(("0.0.0.0", 7860), Handler).serve_forever()