Upload app.py

app.py

@@ -8,8 +8,6 @@ import torch
 import torch.nn as nn
 from glycowork.motif.graph import glycan_to_nxGraph
 import networkx as nx
-import matplotlib.pyplot as plt
-from networkx.drawing.nx_agraph import write_dot
 import pydot
 # import pygraphviz as pgv
 
@@ -46,6 +44,7 @@ def fn(glycan, model):
     labels = {i:node_labels[i] for i in range(len(graph.nodes))}
     graph = nx.relabel_nodes(graph, labels)
     graph = nx.drawing.nx_pydot.to_pydot(graph)
+    graph.set_prog("neato")
     graph.write_png("graph.png")
     # write_dot(graph, "graph.dot")
     # graph=pgv.AGraph("graph.dot")  
@@ -88,7 +87,7 @@ 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.Label(num_top_classes=15, label="Prediction"), gr.Image(label="Glycan graph")],
-    allow_flagging=False,
+    allow_flagging="Never",
     title="SweetNet demo",
     examples=[["GlcOSN(a1-4)GlcA(b1-4)GlcOSN(a1-4)GlcAOS(b1-4)GlcOSN(a1-4)GlcOSN", "No data augmentation"],
     ["Man(a1-2)Man(a1-3)[Man(a1-3)Man(a1-6)]Man(b1-4)GlcNAc(b1-4)GlcNAc", "Random node deletion"],