Update app.py

app.py

@@ -224,10 +224,10 @@ def colorize2(img: Image.Image, model_option: str):
 
     with torch.inference_mode():
         out2 = model_net(input)
-        out = sample[0].cpu().numpy().transpose([1,2,0])
-        out = np.uint8(((out + 1) / 2 * 255).clip(0,255))
+        out2 = sample[0].cpu().numpy().transpose([1,2,0])
+        out2 = np.uint8(((out + 1) / 2 * 255).clip(0,255))
     
-    return Image.fromarray(out).convert('RGB')
+    return Image.fromarray(out2).convert('RGB')
 
 
 with gr.Blocks() as demo: