Update app.py

app.py

@@ -217,11 +217,11 @@ def colorize2(img: Image.Image, model_option: str):
     if not isinstance(img, dict):
         return gr.update(visible=True)
 
-    if hint_mode == "Model 1":
+    if model_option == "Model 1":
         model_int = 0
-    elif hint_mode == "Model 2":
+    elif model_option == "Model 2":
         model_int = 0
-
+    input = torch.from_numpy(np.asarray(img))[None,None].float().to(device) / 255.0 * 2 - 1
     with torch.inference_mode():
         out2 = model_net(input)
         out2 = sample[0].cpu().numpy().transpose([1,2,0])