Upload app.py

app.py

@@ -15,12 +15,12 @@ def show_preds_image(image_path):
     A = (A - 127.5) / 127.5
     A = np.expand_dims(A,axis=0)
     B = model.predict(A)
-    B=np.squeeze(B,axis=0)
+    B = np.squeeze(B,axis=0)
     B = (B + 1) / 2.0
     return B
  
 inputs_image = [
-    gr.components.Image(type="filepath", label="Input Image"),
+    gr.components.Image(shape=(256,256),type="filepath", label="Input Image"),
 ]
 outputs_image = [
     gr.components.Image(type="numpy", label="Output Image"),