fix gradio

app.py

@@ -8,9 +8,9 @@ try:
     router_pipeline = pipeline(
         "text-classification",
         model=MODEL_NAME,
-        return_all_scores=True
+        return_all_scores=True,
+        device='cuda'
     )
-    router_pipeline.to('cuda')
 except Exception as e:
     print(f"Error loading model: {e}")
     router_pipeline = None
@@ -18,7 +18,7 @@ except Exception as e:
 @spaces.GPU
 def classify_query(query_text):
     if not router_pipeline:
-        return {"Error": "Model could not be loaded. Check Space logs for details."}
+        raise gr.Error("Model could not be loaded. Check Space logs for details.")
 
     if not query_text or not query_text.strip():
         return {"Vector Search": 0.0, "Lexical Search": 0.0}
@@ -44,9 +44,9 @@ Enter an e-commerce query to determine if it's better for **vector search** (con
 examples = [
     ["father day gift"],
     ["16x16 pillow cover"],
-    ["something to wear for a wedding"],
-    ["logitech mx master 3s mouse"],
-    ["comfortable office chair"],
+    ["office chair"],
+    ["prom dress"],
+    ["sony wh1000xm4"],
 ]
 
 app = gr.Interface(
@@ -69,3 +69,4 @@ app = gr.Interface(
 
 if __name__ == "__main__":
     app.launch()
+