Update app.py

app.py

@@ -26,7 +26,7 @@ def replace_single_quotes(text):
     return replaced_text
 
 class PromptEnhancerV2:
-    @zero.gpu 
+    @spaces.GPU 
     def __init__(self, models_root_path, device_map="auto", torch_dtype="bfloat16"):#auto
         from transformers import Qwen2_5_VLForConditionalGeneration, AutoProcessor
         if not logging.getLogger(__name__).handlers:
@@ -49,8 +49,7 @@ class PromptEnhancerV2:
         self.processor = AutoProcessor.from_pretrained(models_root_path)
 
     # @torch.inference_mode()
-    # @spaces.GPU
-    @zero.gpu 
+    @spaces.GPU
     def predict(
         self,
         prompt_cot,