Removed mps

app.py

@@ -10,7 +10,6 @@ from transformers import (AutoConfig, AutoModelForObjectDetection, AutoImageProc
 from PIL import Image, ImageDraw, ImageFont
 import pkgutil
 
-
 # Load dataset from Hugging Face
 dataset1 = load_dataset("gcesar/spinach", download_mode="force_redownload")
 dataset2 = load_dataset("gcesar/spinach_augment", download_mode="force_redownload")
@@ -30,16 +29,16 @@ dataset2 = load_dataset("gcesar/spinach_augment", download_mode="force_redownloa
 # torch.backends.mps.is_built()
 
 # Assign GPU
-device = torch.device("mps")
+# device = torch.device("mps")
 
 # Use GPU
 # .to(device)
 # pipeline(device=device)
 
 # Create pipeline model
-#pipe = pipeline(task="object-detection", model="haiquanua/weed_detr")
+pipe = pipeline(task="object-detection", model="haiquanua/weed_detr", trust_remote_code=True, force_download=True)
 # Create pipeline model with mps
-pipe = pipeline(task="object-detection", model="haiquanua/weed_detr", device=device, trust_remote_code=True, force_download=True)
+#pipe = pipeline(task="object-detection", model="haiquanua/weed_detr", device=device, trust_remote_code=True, force_download=True)
 
 # repo_path="haiquanua/weed_swin"
 #