using zerogpu

Signed-off-by: Zhang Jun <[email protected]>

app.py

@@ -9,6 +9,7 @@ import os
 # Set environment variable to avoid tokenizer parallelism deadlocks
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 
+import spaces
 from ocr_model import HunyuanOCR
 from visualization import draw_detection_boxes, get_detection_summary
 from dotenv import load_dotenv
@@ -56,6 +57,7 @@ def initialize_model():
     return ocr_model
 
 
+@spaces.GPU
 def process_image(image: Image.Image, prompt: str = None, target_language: str = "Chinese"):
     """
     Process uploaded image and return annotated result

requirements.txt

@@ -8,3 +8,4 @@ requests>=2.31.0
 openai>=1.0.0
 python-dotenv>=1.0.0
 huggingface_hub>=0.19.0
+spaces>=0.30.0