Upload app.py with huggingface_hub

app.py

@@ -31,6 +31,7 @@ except ImportError:
 import json
 import logging
 import os
+from importlib import metadata, util
 
 import gradio as gr
 from dotenv import load_dotenv
@@ -51,8 +52,35 @@ logger = logging.getLogger(__name__)
 # ---------------------------------------------------------------------------
 IS_SPACES = os.getenv("SPACE_ID") is not None
 
+spaces = None
 if IS_SPACES:
-    import spaces
+    try:
+        import spaces as _spaces
+
+        # Local `spaces/` directory can shadow the HF `spaces` package.
+        if hasattr(_spaces, "GPU"):
+            spaces = _spaces
+        else:
+            raise ImportError("Imported `spaces` module has no GPU decorator")
+    except Exception:
+        try:
+            # Load the installed HF spaces package directly from site-packages.
+            dist = metadata.distribution("spaces")
+            module_path = dist.locate_file("spaces/__init__.py")
+            spec = util.spec_from_file_location("hf_spaces_runtime", module_path)
+            if spec is None or spec.loader is None:
+                raise ImportError("Could not load spaces package spec")
+            _spaces = util.module_from_spec(spec)
+            spec.loader.exec_module(_spaces)
+            if hasattr(_spaces, "GPU"):
+                spaces = _spaces
+            else:
+                raise ImportError("Installed spaces package has no GPU decorator")
+        except Exception as e:
+            logger.warning(
+                "HF Spaces GPU decorator unavailable (%s). Falling back to non-GPU wrappers.",
+                e,
+            )
 
 # ---------------------------------------------------------------------------
 # Custom CSS
@@ -357,13 +385,24 @@ def ask(question: str, api_key: str) -> str:
 # GPU-decorated functions for ZeroGPU
 # ---------------------------------------------------------------------------
 
+def _gpu_wrapper(duration: int):
+    """Use HF ZeroGPU decorator when available; otherwise no-op."""
+
+    def passthrough(fn):
+        return fn
+
+    if IS_SPACES and spaces is not None and hasattr(spaces, "GPU"):
+        return spaces.GPU(duration=duration)
+    return passthrough
+
+
 if IS_SPACES:
-    @spaces.GPU(duration=120)
+    @_gpu_wrapper(duration=120)
     def _classify_gpu(description: str) -> dict:
         """Classify incident using the fine-tuned model on ZeroGPU."""
         return classify_incident(description=description, use_hf=True)
 
-    @spaces.GPU(duration=120)
+    @_gpu_wrapper(duration=120)
     def _ask_gpu(question: str) -> str:
         """Answer question using the fine-tuned model on ZeroGPU."""
         return answer_question(question=question, use_hf=True)