Update app.py

app.py

@@ -7,8 +7,14 @@ from PIL import Image
 from pytorch_grad_cam import GradCAM
 from pytorch_grad_cam.utils.image import show_cam_on_image
 import gradio as gr
+from huggingface_hub import hf_hub_download
+import yolo9
 
-model = torch.load('best.pt')
+model = yolo9.load('best.pt')
+
+# Set model parameters
+model.conf = conf_threshold
+model.iou = iou_threshold
 
 classes = ('ball', 'goalkeeper', 'player', 'referee')