Update app.py

app.py

@@ -18,6 +18,7 @@ sess = tf.compat.v1.Session()
 from keras import backend as K
 K.set_session(sess)
 
+graph = tf.get_default_graph() 
 
 # Do you want it loud? 
 VERBOSE = 1
@@ -87,7 +88,8 @@ class Network(object):
       other_data_input = data_input.reshape((self.G, self.G, self.G), order='F')
 
       # Get the outputs
-      predicted_output = self.network.predict(data_input)
+      with graph.as_default():
+          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))
 
@@ -112,7 +114,8 @@ class Network(object):
       other_data_input = data_input.reshape((3, self.F))
 
       # Get the outputs
-      predicted_output = self.network.predict(data_input)
+      with graph.as_default():
+          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')
 
@@ -125,7 +128,8 @@ class Network(object):
       data_input = other_data_input.reshape((1, 3*self.F))
  
       # Get the outputs
-      predicted_output = self.network.predict(data_input)
+      with graph.as_default():
+          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