Update app.py

app.py

@@ -4,8 +4,6 @@ from PIL import Image
 import json
 import spaces
 
-@spaces.GPU
-
 # === Model Mapping ===
 MODEL_MAP = {
     "Qwen": "models.qwen",
@@ -159,7 +157,7 @@ def process_single_image(model_name, image_file):
 
     return image, json_output, pretty_output
 
-# === Gradio Interface
+# === Gradio UI
 iface = gr.Interface(
     fn=process_single_image,
     inputs=[
@@ -176,7 +174,10 @@ iface = gr.Interface(
     allow_flagging="never"
 )
 
+# === Enable GPU mode and launch
+@spaces.GPU
+def main():
+    iface.launch()
+
 if __name__ == "__main__":
     main()
-
-#iface.launch(share=True)