Update app.py

Add webcam capture, keep original resolution as default model input (remove resize)

app.py

@@ -83,20 +83,15 @@ def infer(image: Image.Image, want_metrics: bool, sizing_mode: str, show_L: bool
     w,h = pil.size
     was_color = not is_grayscale(pil)
 
-    if sizing_mode == "Pad to keep size":
-        proc, (oh, ow) = pad_to_multiple(rgb, 16); back = (ow, oh)
-    else:
-        proc = cv2.resize(rgb, (256,256), interpolation=cv2.INTER_CUBIC); back = (w,h)
+    
+    proc, (oh, ow) = pad_to_multiple(rgb, 16); back = (ow, oh)
 
     L = to_L(proc)
     with torch.no_grad():
         ab = model(L)
     out = lab_to_rgb(L, ab)
 
-    if sizing_mode == "Pad to keep size":
-        out = out[:back[1], :back[0]]
-    else:
-        out = cv2.resize(out, back, interpolation=cv2.INTER_CUBIC)
+    out = out[:back[1], :back[0]]
 
     # Metrics (Gradio-native numbers)
     mae = psnr = ssim = None
@@ -152,24 +147,19 @@ def make_theme():
 THEME = make_theme()
 
 # ---------------- UI ----------------
-with gr.Blocks(theme=THEME, title="Neural Colorizer") as demo:
-    gr.Markdown("# 🎨 Neural Colorizer")
+with gr.Blocks(theme=THEME, title="Image Colorizer") as demo:
+    gr.Markdown("# 🎨 Image Colorizer")
 
     with gr.Row():
         with gr.Column(scale=5):
             img_in = gr.Image(
-                label="Upload grayscale or color image",
+                label="Upload grayscale image",
                 type="pil",
                 image_mode="RGB",
                 height=320,
-                sources=["upload", "clipboard"]
+                sources=["upload", "webcam", "clipboard"]
             )
             with gr.Row():
-                sizing = gr.Radio(
-                    ["Resize to 256", "Pad to keep size"],
-                    value="Resize to 256",
-                    label="Sizing"
-                )
                 show_L = gr.Checkbox(label="Show L-channel", value=False)
             show_m = gr.Checkbox(label="Show metrics", value=True)
             with gr.Row():
@@ -199,14 +189,14 @@ with gr.Blocks(theme=THEME, title="Neural Colorizer") as demo:
             extras  = gr.HTML()
 
     def _go(image, want_metrics, sizing_mode, show_L):
-        o, c, mae, psnr, ssim, cmp_html, extra = infer(image, want_metrics, sizing_mode, show_L)
+        o, c, mae, psnr, ssim, cmp_html, extra = infer(image, want_metrics, show_L)
         if not want_metrics:
             mae = psnr = ssim = None
         return o, c, mae, psnr, ssim, cmp_html, extra
 
     run.click(
         _go,
-        inputs=[img_in, show_m, sizing, show_L],
+        inputs=[img_in, show_m, show_L],
         outputs=[orig, out, mae_box, psnr_box, ssim_box, compare, extras]
     )