Update app.py

app.py

@@ -248,7 +248,7 @@ label_mapping = [
 MODEL_NAME = "valentinafeve/yolos-fashionpedia"
 
 feature_extractor = YolosImageProcessor.from_pretrained('hustvl/yolos-small')
-model = YolosForObjectDetection.from_pretrained(MODEL_NAME).to(device)
+model = YolosForObjectDetection.from_pretrained(MODEL_NAME)
 
 def get_category_index(category):
     # Find index of label mapping
@@ -380,7 +380,7 @@ def get_objects(image, threshold=0.8):
     image = image.resize((600, 800))
 
     inputs = feature_extractor(images=image, return_tensors="pt")
-    outputs = model(**inputs.to(device))
+    outputs = model(**inputs)
 
     detected_classes, output_img, crops, crop_classes = visualize_predictions(image, outputs, threshold=threshold)
     for cl in detected_classes: