Base application

app.py

@@ -295,4 +295,4 @@ with demo:
             gr.Textbox(value=CITATION_BUTTON_TEXT, label=CITATION_BUTTON_LABEL, lines=8, show_copy_button=True)
 
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    demo.launch(server_name="0.0.0.0", server_port=7860, ssr=False)

src/submission/submit.py

@@ -7,6 +7,20 @@ import threading
 import traceback
 from datetime import datetime, timezone
 
+# Hugging Face ZeroGPU compatibility
+try:
+    import spaces
+    has_spaces = True
+except ImportError:
+    has_spaces = False
+
+def gpu_decorator(duration=120):
+    def decorator(fn):
+        if has_spaces and hasattr(spaces, "GPU"):
+            return spaces.GPU(duration=duration)(fn)
+        return fn
+    return decorator
+
 # Engine import
 try:
     from src.audit.engine import AuditError, check_feasibility, run_full_audit
@@ -99,7 +113,6 @@ def clean_model_name(raw_name: str) -> str:
 
     # Strip HTML tags
     name = re.sub(r'<[^>]+>', '', name)
-    # Strip markdown label brackets [Model Name] -> Model Name
     name = re.sub(r'\[([^\]]+)\]', r'\1', name)
     name = name.replace("https://huggingface.co/", "")
     return name.strip().strip("/")
@@ -130,6 +143,7 @@ def _save_cert(model_id: str, revision: str, cert: dict) -> str:
     return out_path
 
 
+@gpu_decorator(duration=120)
 def execute_direct_xray_audit(model_id: str, revision: str = "main", trust_remote_code: bool = False, progress_callback=None) -> dict:
     now_str = datetime.now(timezone.utc).strftime("%Y-%m-%d %H:%M:%S")