Update app.py

app.py

@@ -62,9 +62,16 @@ class SDRecommender:
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
         print(f"Using device: {self.device}")
         
-        # Load Florence model and processor
-        self.processor = AutoProcessor.from_pretrained("microsoft/Florence-2-large")
-        self.florence = AutoModel.from_pretrained("microsoft/Florence-2-large")
+        # Load Florence model and processor dengan trust_remote_code=True
+        print("Loading Florence model and processor...")
+        self.processor = AutoProcessor.from_pretrained(
+            "microsoft/Florence-2-large",
+            trust_remote_code=True
+        )
+        self.florence = AutoModel.from_pretrained(
+            "microsoft/Florence-2-large",
+            trust_remote_code=True
+        )
         
         # Load dataset
         print("Loading dataset...")
@@ -85,7 +92,7 @@ class SDRecommender:
         
         # Load trained weights if available
         if os.path.exists("recommender_model.pth"):
-            self.model.load_state_dict(torch.load("recommender_model.pth"))
+            self.model.load_state_dict(torch.load("recommender_model.pth", map_location=self.device))
             print("Loaded trained model weights")
         self.model.eval()