Update app.py

app.py

@@ -25,7 +25,7 @@ ANNOTATION_THICKNESS = 4     # Thickness of bounding box lines
 repo_id = 'edeler/ICC'  # Replace with your Hugging Face repository ID
 model_dir = snapshot_download(repo_id, local_dir='./models/ICC')
 model_path = os.path.join(model_dir, "best.pt")  # Adjust if filename differs
-model = YOLO(model_path).to(cuda)
+model = YOLO(model_path).to(device)
 
 # Define the detection function for Gradio
 def detect_objects(image: np.ndarray) -> Image.Image: