Update app.py

app.py

@@ -96,9 +96,9 @@ def predict(input_image):
 
 iface = gr.Interface(
     fn=predict,
-    inputs=gr.Image(label="Upload molecular image", type="pil", show_label=False).style(height=256),
+    inputs=gr.Image(label="Upload molecular image", type="pil", show_label=False),
     outputs=[
-        gr.Image(label="Prediction").style(height=256),
+        gr.Image(label="Prediction"),
         gr.Text(label="SMILES").style(show_copy_button=True),
     ],
     title="OCSR",