Add Spaces GPU decorators for ZeroGPU startup

app.py

@@ -8,6 +8,18 @@ import config
 import utils
 import pandas as pd
 
+try:
+    import spaces
+except Exception:
+    spaces = None
+
+
+def _gpu_decorator(func):
+    """Use Spaces GPU decorator when available; no-op locally."""
+    if spaces is None:
+        return func
+    return spaces.GPU(func)
+
 # ===================== UTILITIES =====================
 
 def load_sample_texts():
@@ -71,6 +83,7 @@ def get_embeddings_examples():
 
 # ===================== SENTIMENT ANALYSIS =====================
 
+@_gpu_decorator
 def demo_sentiment(text):
     """Demo sentiment analysis."""
     if not text.strip():
@@ -82,6 +95,7 @@ def demo_sentiment(text):
 
 # ===================== NER =====================
 
+@_gpu_decorator
 def demo_ner(text):
     """Demo named entity recognition."""
     if not text.strip():
@@ -95,6 +109,7 @@ def demo_ner(text):
 
 # ===================== QUESTION ANSWERING =====================
 
+@_gpu_decorator
 def demo_qa(context, question):
     """Demo question answering."""
     if not context.strip() or not question.strip():
@@ -108,6 +123,7 @@ def demo_qa(context, question):
 
 # ===================== SUMMARIZATION =====================
 
+@_gpu_decorator
 def demo_summarization(text):
     """Demo text summarization."""
     if not text.strip():
@@ -124,6 +140,7 @@ def demo_summarization(text):
 
 # ===================== SEMANTIC SIMILARITY =====================
 
+@_gpu_decorator
 def demo_similarity(text1, text2):
     """Demo semantic similarity."""
     if not text1.strip() or not text2.strip():
@@ -137,6 +154,7 @@ def demo_similarity(text1, text2):
 
 # ===================== TOKENIZATION =====================
 
+@_gpu_decorator
 def demo_tokenization(text):
     """Demo tokenization."""
     if not text.strip():