Update app.py

app.py

@@ -6,7 +6,8 @@ import os
 import spaces
 
 
-model = YOLOv10.from_pretrained('jameslahm/yolov10x')
+device = 'cuda' if torch.cuda.is_available() else 'cpu'
+model = YOLOv10.from_pretrained('jameslahm/yolov10x').to(device)
 
 # Define activity categories based on detected objects
 activity_categories = {
@@ -32,7 +33,6 @@ def categorize_activity(detected_objects):
 # Function to process the video, detect objects, and generate a categorized journal with images
 @spaces.GPU
 def generate_journal_with_images(video_path):
-    model.to("cuda")
     cap = cv2.VideoCapture(video_path)
     journal_entries = {}
     saved_images = []