Update app.py

app.py

@@ -223,7 +223,7 @@ def colorize2(img: Image.Image, model_option: str):
         model_int = 0
 
     with torch.inference_mode():
-        out2 = model(input)
+        out2 = model_net(input)
         out = sample[0].cpu().numpy().transpose([1,2,0])
         out = np.uint8(((out + 1) / 2 * 255).clip(0,255))