Update app.py

app.py

@@ -5,10 +5,10 @@ import gradio as gr
 from ultralytics import YOLO
 
 # Check if CUDA (GPU support) is available
-device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+# device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 # Load the YOLOv8 model
-model = YOLO('yolov8x-seg.pt').to(device)
+model = YOLO('yolov8x-seg.pt')
 
 def process_video(input_video_path):
     cap = cv2.VideoCapture(input_video_path)
@@ -31,7 +31,7 @@ def process_video(input_video_path):
         resized_frame = cv2.resize(frame, (640, 640))
 
         # Convert resized frame to torch tensor and move it to GPU
-        frame_tensor = torch.from_numpy(resized_frame).permute(2, 0, 1).unsqueeze(0).float().to(device) / 255.0
+        frame_tensor = torch.from_numpy(resized_frame).permute(2, 0, 1).unsqueeze(0).float() / 255.0
 
         threshold = 0.1
         frame_copy = frame.copy()