Update app.py

app.py

@@ -3,6 +3,8 @@
 Doubles as a portfolio piece (preloaded contract clauses) and a sales tool
 (paste any documents and search them).
 """
+import spaces  # must be imported before torch / CUDA-touching packages
+
 import hashlib
 
 import gradio as gr
@@ -62,6 +64,7 @@ def _embed_docs(docs):
     return _cache[key]
 
 
+@spaces.GPU
 def search(documents_text, query, top_k):
     docs = [d.strip() for d in (documents_text or "").split("\n\n") if d.strip()]
     if not docs:
@@ -90,7 +93,7 @@ def search(documents_text, query, top_k):
     return "\n".join(out)
 
 
-with gr.Blocks(title="Semantic Document Search", theme=gr.themes.Soft()) as demo:
+with gr.Blocks(title="Semantic Document Search") as demo:
     gr.Markdown(
         """
         # 🔍 Semantic Document Search