Update app.py

app.py

@@ -10,7 +10,7 @@ class LeffaPredictor:
     def __init__(self):
         # Carica i pesi del modello
         # Sostituisci 'weights.pth' con il file del modello Leffa
-        self.model = torch.load("weights.pth", map_location=torch.device('cpu'))
+        self.model = torch.load("virtual_tryon_dc.pth", map_location=torch.device('cpu'))
 
     def leffa_predict_vt(self, human_img_path, garment_img_path, garment_type):
         # Simula il processo del modello (sostituisci con la logica reale)