Update app.py

app.py

@@ -24,11 +24,8 @@ def print_bn():
 
 def update_bn(image):
     cursor_im = 0
-    print("Before Resize:", image.shape)
-    
     image = T.Resize((40,40))(image)
     image = image.reshape(-1)
-    print("After Resize:", image.shape)
     for m in model.modules():
         if(type(m) is nn.BatchNorm2d):
             if(cursor_im < image.shape[0]):
@@ -44,10 +41,6 @@ def update_bn(image):
     
 
 def greet(image):
-    # url = f'https://huggingface.co/spaces?p=1&sort=modified&search=GPT'
-    # html = request_url(url)
-    # key = os.getenv("OPENAI_API_KEY")
-#     x = torch.ones([1,3,224,224])
     if(image is None):
         bn_data = print_bn()
         return ','.join([f'{x:.2f}' for x in bn_data])
@@ -61,11 +54,11 @@ def greet(image):
         update_bn(image)
         print(image.shape)
         out = model(image)
-    # model.train()
     return "Hello world!"
 
 
 
-image = gr.inputs.Image(label="Upload a photo for beautify", shape=(224,224))
-iface = gr.Interface(fn=greet, inputs=image, outputs="text")
+image = gr.inputs.Image(label="Upload a photo for beauty", shape=(224,224))
+out_image = gr.inputs.Image(label='Yes, it becomes better.')
+iface = gr.Interface(fn=greet, inputs=image, outputs='text')
 iface.launch()