Update app.py

app.py

@@ -2,7 +2,7 @@
 # 1. Import the required dependencies
 import gradio as gr
 import torch 
-# import spaces # for GPU usage
+import spaces # for GPU usage
 
 from PIL import Image, ImageDraw, ImageFont 
 
@@ -44,7 +44,7 @@ color_dict = {
 
 # 3. Create a function predict_on_image
 # Use a GPU on a target function
-# @spaces.GPU
+@spaces.GPU
 def predict_on_image(image, conf_threshold):
   model.eval()