Upload app.py

app.py

@@ -82,7 +82,7 @@ def fn(glycan, model):
 demo = gr.Interface(
     fn=fn,
     inputs=[gr.Textbox(label="Glycan sequence"), gr.Radio(label="Model",choices=["No data augmentation", "Random node deletion", "Ensemble"])],
-    outputs=[gr.Image(label="Glycan graph", shape=(128,128)), gr.Label(num_top_classes=15, label="Prediction")],
+    outputs=[gr.Image(label="Glycan graph"), gr.Label(num_top_classes=15, label="Prediction")],
     allow_flagging=False,
     title="SweetNet demo",
     examples=[["GlcOSN(a1-4)GlcA(b1-4)GlcOSN(a1-4)GlcAOS(b1-4)GlcOSN(a1-4)GlcOSN", "No data augmentation"],