fix: revert to a previous model

app.py

@@ -2,15 +2,16 @@ import gradio as gr
 import torch
 from PIL import Image
 import numpy as np
+from huggingface_hub import hf_hub_download
+
 import cv2
 
 # Download the model from Hugging Face Hub
-model_path = 'latest.pt'
+model_path = hf_hub_download(repo_id='ethandavey/yoloV5-coursework-model', filename='model.pt')
 
 # Load the YOLOv5 model
 model = torch.hub.load('ultralytics/yolov5', 'custom', path=model_path)
 model.line_thickness = 3  # Set the thickness of the bounding box lines
-model.conf = 0.001  # Set the confidence threshold for detections
 
 # Print model class names
 print("Model class names:", model.names)