Update app.py

app.py

@@ -1,3 +1,5 @@
+'''
+
 import gradio as gr
 import os
 if os.environ.get("SPACES_ZERO_GPU") is not None:
@@ -43,7 +45,6 @@ demo.launch()
 
 
 
-'''
 
 import gradio as gr
 from huggingface_hub import InferenceClient
@@ -376,6 +377,8 @@ try:
 except Exception as e:
     print(f"[ERROR] Workflow execution failed: {e}")
 
+'''
+
 from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
 from langchain_huggingface import HuggingFacePipeline
 from langchain.tools import Tool
@@ -383,6 +386,20 @@ from langchain.agents import create_react_agent
 from langgraph.graph import StateGraph, END
 from pydantic import BaseModel
 import gradio as gr
+import os
+if os.environ.get("SPACES_ZERO_GPU") is not None:
+    import spaces
+else:
+    class spaces:
+        @staticmethod
+        def GPU(func):
+            def wrapper(*args, **kwargs):
+                return func(*args, **kwargs)
+            return wrapper
+
+@spaces.GPU
+def fake_gpu():
+  pass
 
 # ---------------------------------------
 # Step 1: Define Hugging Face LLM (Qwen/Qwen2.5-7B-Instruct-1M)
@@ -603,5 +620,3 @@ iface = gr.Interface(
 if __name__ == "__main__":
     iface.launch()
 
-
-'''