Update app.py

app.py

@@ -152,7 +152,7 @@ def preprocess_image(image):
 
     return transform(image).unsqueeze(0)
     
-@spaces.GPU
+@spaces.GPU()
 async def predict_single_dog(image):
     """
     Predicts the dog breed using only the classifier.
@@ -185,7 +185,7 @@ async def predict_single_dog(image):
         return probabilities[0], breeds[:3], relative_probs
         
 
-@spaces.GPU
+@spaces.GPU()
 async def detect_multiple_dogs(image, conf_threshold=0.3, iou_threshold=0.55):
     results = model_yolo(image, conf=conf_threshold, iou=iou_threshold)[0]
     dogs = []
@@ -268,7 +268,7 @@ def create_breed_comparison(breed1: str, breed2: str) -> dict:
     return comparison_data
     
 
-@spaces.GPU
+@spaces.GPU()
 async def predict(image):
     """
     Main prediction function that handles both single and multiple dog detection.