added back pickle
Browse files
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(
|
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 |
|