Update app.py

app.py

@@ -143,7 +143,7 @@ class ECCVGenerator(BaseColor):
 
 # model_net = torch.load(f'weights/colorizer.pt')
 model_net = ECCVGenerator()
-model_net.load_state_dict(torch.load(f'weights/colorizer.pt'))
+model_net.load_state_dict(torch.load(f'weights/colorizer (1).pt'))
 
 
 def resize_original(img: Image.Image):
@@ -171,6 +171,17 @@ def resize_original(img: Image.Image):
     return gr.Image.update(value=guide_img.convert('RGBA')), guide_img.convert('RGBA')
 
 
+def resize_original2(img: Image.Image):
+    if img is None:
+        return img
+    if isinstance(img, dict):
+        img = img["image"]
+    
+    img = img.resize(256,256)
+
+    return img
+
+
 def colorize(img: Dict[str, Image.Image], guide_img: Image.Image, seed: int, hint_mode: str):
     if not isinstance(img, dict):
         return gr.update(visible=True)