Update app.py

app.py

@@ -26,7 +26,8 @@ os.makedirs(DATA_DIR, exist_ok=True)
 os.makedirs(MODEL_DIR, exist_ok=True)
 os.makedirs(META_DIR, exist_ok=True)
 
-DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+# Force CPU on Hugging Face Spaces for this lightweight demo
+DEVICE = torch.device("cpu")
 CLASS_NAMES = [str(i) for i in range(10)]
 
 
@@ -513,4 +514,4 @@ with gr.Blocks(title="Image Classification") as demo:
 
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(ssr_mode=False)