Update app.py

app.py

@@ -1,20 +1,13 @@
 import gradio as gr
-import torch
-from PIL import Image
-import torchvision.transforms as T
+# import torch
+# from PIL import Image
+# import torchvision.transforms as T
 from ultralytics import YOLO
 import cv2
 import numpy as np
 
 # Load the PT model
 model = YOLO("Model_IV.pt")
-# checkpoint = torch.load("Model_IV.pt")
-
-# # Define preprocessing
-# transform = T.Compose([
-#     T.Resize((224, 224)),  # Adjust to your model's input size
-#     T.ToTensor(),
-# ])
 
 def predict(image):
     # Preprocessing: Convert the colour space to RGB