first commit

app.py

@@ -46,14 +46,16 @@ class CUTModel:
         self.model.netG.load_state_dict(state_dict)  # Load state_dict langsung ke netG
         self.model.eval()
         
-        # Transformasi input
+        # Transformasi input - Tanpa resize
         self.transform = transforms.Compose([
-            transforms.Resize((256, 256)),
             transforms.ToTensor(),
             transforms.Normalize((0.5,), (0.5,))
         ])
 
     def predict(self, image):
+        # Simpan ukuran asli
+        original_size = image.size
+
         # Preprocess
         image_tensor = self.transform(image).unsqueeze(0).to(self.device)
         
@@ -61,9 +63,10 @@ class CUTModel:
         with torch.no_grad():
             output = self.model.netG(image_tensor)
         
-        # Postprocess
+        # Postprocess - Resize ke ukuran asli
         output_image = (output.squeeze().cpu().clamp(-1, 1) + 1) / 2.0
-        return transforms.ToPILImage()(output_image)
+        output_image = transforms.ToPILImage()(output_image).resize(original_size)
+        return output_image
 
 # Inisialisasi model
 print("Inisialisasi model...")