Upload app.py

app.py

@@ -10,9 +10,9 @@ import numpy as np
 import torch
 import torch.nn as nn
 from networkx.drawing.nx_agraph import write_dot
-import pygraphviz as pgv
-from glycowork.motif.graph import glycan_to_nxGraph
-import networkx as nx
+# import pygraphviz as pgv
+# from glycowork.motif.graph import glycan_to_nxGraph
+# import networkx as nx
 
 class EnsembleModel(nn.Module):
     def __init__(self, models):
@@ -41,14 +41,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":