jpterry commited on
Commit
a78b672
·
1 Parent(s): 495e4b4

added back pickle

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -137,7 +137,7 @@ def predict_and_analyze(model_name, num_channels, dim, image):
137
  # im = f.readlines()
138
  # image = np.frombuffer(image)
139
 
140
- image = np.load(image)
141
 
142
  image = image.reshape((num_channels, W, W))
143
 
@@ -182,7 +182,7 @@ demo = gr.Interface(
182
  gr.Image(label="Actication 2", show_label=True)],
183
  title="Kinematic Planet Detector"
184
  )
185
- demo.launch(share=True)
186
 
187
 
188
 
 
137
  # im = f.readlines()
138
  # image = np.frombuffer(image)
139
 
140
+ image = np.load(image, allow_pickle=True)
141
 
142
  image = image.reshape((num_channels, W, W))
143
 
 
182
  gr.Image(label="Actication 2", show_label=True)],
183
  title="Kinematic Planet Detector"
184
  )
185
+ demo.launch()
186
 
187
 
188