Update app.py

app.py

@@ -5,9 +5,8 @@ import torch
 import os
 import spaces
 
-# Load YOLOv10 model with CUDA if available
-device = 'cuda' if torch.cuda.is_available() else 'cpu'
-model = YOLOv10.from_pretrained('jameslahm/yolov10x').to(device)
+
+model = YOLOv10.from_pretrained('jameslahm/yolov10x')
 
 # Define activity categories based on detected objects
 activity_categories = {
@@ -33,6 +32,7 @@ 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 = []