wip - zero gpu test

app.py

@@ -1,13 +1,24 @@
 # Standard library imports
 import os
 import gradio as gr
+import spaces
+
+
 
 # Local imports
-from age_estimation.age_estimation import age_estimation
-from detection.face_detection import face_detection
-from detection.object_detection import object_detection
+from age_estimation.age_estimation import age_estimation as _age_estimation
+from detection.face_detection import face_detection as _face_detection 
+from detection.object_detection import object_detection as _object_detection
 from utils.ui_utils import update_input_visibility
 
+@spaces.GPU
+def age_estimation(input_type, uploaded_image, image_url, base64_string):
+    return _age_estimation(input_type, uploaded_image, image_url, base64_string)
+
+@spaces.GPU
+def object_detection(input_type, uploaded_image, image_url, base64_string):
+    return _object_detection(input_type, uploaded_image, image_url, base64_string)
+
 with gr.Blocks() as demo:
     # Add a title to the interface
     gr.Markdown("# Computer Vision Tools")