Add ZeroGPU

app.py

@@ -3,6 +3,7 @@ import os.path
 from io import BytesIO
 from pathlib import Path
 
+import spaces
 import glob
 import numpy as np
 import gradio as gr
@@ -156,8 +157,8 @@ def _bar_chart(top_scores, img_name, cmap) -> str:
     b64 = base64.b64encode(buf.getvalue()).decode()
     return f'<img src="data:image/png;base64,{b64}" style="display:block;margin:auto;width:{IMG_PX}px;" />'
 
-# import spaces
-# @spaces.GPU  # ZeroGPU does not seem to be working
+
+@spaces.GPU
 def classify(images, class_text):
     class_names = [c.strip() for c in class_text.split(",") if c.strip()]
     cards = []