add zerogpu

app.py

@@ -1,4 +1,5 @@
 import os
+import spaces
 import re
 import random
 import gradio as gr
@@ -10,7 +11,6 @@ from agent_tools import RestorationToolkit
 
 # download tools ckpts
 from huggingface_hub import hf_hub_download, list_files_info
-import os
 
 # 目标目录
 target_dir = "JarvisIR/checkpoints/agent_tools"
@@ -198,6 +198,7 @@ def resize_image_to_original(processed_image_path, original_size):
         return output_path
     return processed_image_path
 
+@spaces.GPU(duration=150)
 def get_llm_response_streaming(image_path):
     """
     Get streaming response from LLM for image analysis
@@ -235,6 +236,7 @@ def get_llm_response_streaming(image_path):
     
     return streamer
 
+@spaces.GPU(duration=150)
 def process_image_with_tools(image_path, models, original_size):
     """
     Process image using the tool engine and restore to original size