Update app.py

app.py

@@ -9,6 +9,11 @@ from firecrawl import FirecrawlApp
 import gradio as gr
 import os
 
+if torch.cuda.is_available():
+    device = "cuda:0"
+else:
+    device = "cpu"
+
 # Initialize LLM and Tools
 
 # local_llm = 'llama3.1'
@@ -35,6 +40,7 @@ google_search = GoogleSearchAPIWrapper()
 
 firecrawl_app = FirecrawlApp(api_key=os.getenv('FIRECRAWL_KEY'))
 
+@spaces.GPU
 # Query Transformation
 query_prompt = PromptTemplate(
     template="""