Update app.py

app.py

@@ -1,21 +1,21 @@
+# 1. CRITICAL: Import spaces FIRST, before any CUDA-related packages
+import spaces
+
+# 2. Now it is safe to import other libraries
 import gradio as gr
 from fastapi import FastAPI
 from app.main import app as fastapi_app
 
-# 1. Import spaces and define a dummy GPU function to pass the startup check
-try:
-    import spaces
-    @spaces.GPU
-    def _hf_zero_gpu_probe():
-        return "ok"
-except ImportError:
-    pass # Safe fallback for local testing
+# 3. Dummy GPU probe function to satisfy ZeroGPU's startup check
+@spaces.GPU
+def _hf_zero_gpu_probe():
+    return "ok"
 
-# 2. Your existing Gradio wrapper
+# 4. Your Gradio wrapper
 with gr.Blocks() as demo:
     gr.Markdown("# 🛡️ Moderation API")
     gr.Markdown("This is a live API. Send POST requests to `/api/v1/moderate`.")
     gr.Markdown("You can view the interactive API docs at [here](/docs) once it starts.")
 
-# 3. Mount FastAPI onto Gradio
+# 5. Mount FastAPI onto Gradio
 app = gr.mount_gradio_app(fastapi_app, demo, path="/")