ccm commited on
Commit
be8288e
·
1 Parent(s): a198fec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -129,7 +129,7 @@ class Network(object):
129
  predicted_output = predicted_output.reshape((self.G, self.G, self.G), order='F')
130
 
131
  # return idx, other_data_input, true_output, predicted_output
132
- return true_output
133
 
134
  def get_geometry(self, idx=None):
135
 
 
129
  predicted_output = predicted_output.reshape((self.G, self.G, self.G), order='F')
130
 
131
  # return idx, other_data_input, true_output, predicted_output
132
+ return predicted_output
133
 
134
  def get_geometry(self, idx=None):
135