Update app.py

app.py

@@ -173,7 +173,7 @@ def init_weights(net, init='norm', gain=0.02):
             nn.init.constant_(m.bias.data, 0.)
             
     net.apply(init_func)
-    print(f"model initialized with {init} initialization")
+    # print(f"model initialized with {init} initialization")
     return net
 
 def init_model(model, device):
@@ -314,7 +314,7 @@ def visualize(model, data, dims):
         
 def log_results(loss_meter_dict):
     for loss_name, loss_meter in loss_meter_dict.items():
-        print(f"{loss_name}: {loss_meter.avg:.5f}")
+        # print(f"{loss_name}: {loss_meter.avg:.5f}")
 
 # pip install fastai==2.4
 from fastai.vision.learner import create_body
@@ -379,7 +379,7 @@ description = "Gradio demo for Image Colorization project. You can give an image
 gr.Interface(
     main_func, 
     [gr.Image(type="filepath", label="Input Image") ], 
-    [gr.Textbox(label="Image Size", height=300),gr.Image(type="pil", label="Output Image"),"image",gr.Image( label="Output Image2"),"image"],
+    [gr.Textbox(label="Image Size"),gr.Image(type="pil", label="Output Image"),"image",gr.Image( label="Output Image2"),"image"],
     title=title,
     description=description,
     examples=[ ['face.jpg'], ['montains.jpg'],['tree.jpg']  ])