Update app

preprocess/inference_preprocess.py

@@ -13,7 +13,6 @@ from scenedetect.detectors import ContentDetector
 
 from scipy.interpolate import interp1d
 from scipy import signal
-import torch
 
 from ultralytics import YOLO
 
@@ -166,7 +165,6 @@ def crop_video(opt, track, cropfile, tight_scale=1):
 
 	return {'track': track, 'proc_track': dets}
 
-@spaces.GPU(duration=150)
 def inference_video(opt, padding=0):
 	videofile = os.path.join(opt.avi_dir, 'video.avi')
 	vidObj = cv2.VideoCapture(videofile)
@@ -186,7 +184,7 @@ def inference_video(opt, padding=0):
 			image_np = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
 
 			# Perform person detection
-			results = yolo_model(image_np.to(device), verbose=False)
+			results = yolo_model(image_np, verbose=False)
 			detections = results[0].boxes
 
 			dets.append([])