Update app.py

app.py

@@ -18,9 +18,6 @@ sess = tf.compat.v1.Session()
 from keras import backend as K
 K.set_session(sess)
 
-global graph
-graph = tf.compat.v1.get_default_graph() 
-
 # Do you want it loud? 
 VERBOSE = 1
 
@@ -89,8 +86,7 @@ class Network(object):
       other_data_input = data_input.reshape((self.G, self.G, self.G), order='F')
 
       # Get the outputs
-      with graph.as_default():
-          predicted_output = self.network.predict(data_input)
+      predicted_output = self.network.predict(data_input)
       true_output = self.new_curves[idx].reshape((3, self.F))
       predicted_output = predicted_output.reshape((3, self.F))
 
@@ -115,8 +111,7 @@ class Network(object):
       other_data_input = data_input.reshape((3, self.F))
 
       # Get the outputs
-      with graph.as_default():
-          predicted_output = self.network.predict(data_input)
+      predicted_output = self.network.predict(data_input)
       true_output = self.new_geometry[idx].reshape((self.G, self.G, self.G), order='F')
       predicted_output = predicted_output.reshape((self.G, self.G, self.G), order='F')
 
@@ -129,8 +124,7 @@ class Network(object):
       data_input = other_data_input.reshape((1, 3*self.F))
  
       # Get the outputs
-      with graph.as_default():
-          predicted_output = self.network.predict(data_input)
+      predicted_output = self.network.predict(data_input)
       predicted_output = predicted_output.reshape((self.G, self.G, self.G), order='F')
 
       # return idx, other_data_input, true_output, predicted_output