Update handler.py

handler.py

@@ -41,7 +41,7 @@ class Inference:
         for name, param in self.ae.named_parameters():
             param.requires_grad = False
         self.unet = Model(T=T, filters=[64,128,256,512], t_dim=t_dim, depth=depth, LDM=LDM)
-        self.unet.load_state_dict(torch.load("unet.pt", weights_only=False, map_location=torch.device('cpu')))
+        self.unet.load_state_dict(torch.load("repository/unet.pt", weights_only=False, map_location=torch.device('cpu')))
         self.unet.eval()
         for name, param in self.unet.named_parameters():
             param.requires_grad = False