Upload app.py

app.py

@@ -40,14 +40,14 @@ model3 = torch.load("model3.pt", map_location=torch.device('cpu'))
 
 def fn(glycan, model):
     # Draw graph
-    graph = glycan_to_nxGraph(glycan)
-    node_labels = nx.get_node_attributes(graph, 'string_labels')
-    labels = {i:node_labels[i] for i in range(len(graph.nodes))}
-    graph = nx.relabel_nodes(graph, labels)
-    write_dot(graph, "graph.dot")
-    graph=pgv.AGraph("graph.dot")  
-    graph.layout(prog='dot')
-    graph.draw("graph.png")
+    # graph = glycan_to_nxGraph(glycan)
+    # node_labels = nx.get_node_attributes(graph, 'string_labels')
+    # labels = {i:node_labels[i] for i in range(len(graph.nodes))}
+    # graph = nx.relabel_nodes(graph, labels)
+    # write_dot(graph, "graph.dot")
+    # graph=pgv.AGraph("graph.dot")  
+    # graph.layout(prog='dot')
+    # graph.draw("graph.png")
     # Perform inference
     if model == "No data augmentation":
       model_pred = model1