Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -35,8 +35,8 @@ def predict(filename, model=None, device=None, reduce_labels=True):
|
|
35 |
for i in range(3):
|
36 |
img_channel_i = img_data[:,:,i]
|
37 |
if len(img_channel_i[np.nonzero(img_channel_i)])>0:
|
38 |
-
|
39 |
-
|
40 |
my_model = MultiStreamCellSegModel.from_pretrained("Lewislou/cellseg_sribd")
|
41 |
checkpoints = torch.load('model.pt')
|
42 |
my_model.__init__(ModelConfig())
|
|
|
35 |
for i in range(3):
|
36 |
img_channel_i = img_data[:,:,i]
|
37 |
if len(img_channel_i[np.nonzero(img_channel_i)])>0:
|
38 |
+
pre_img_data[:,:,i] = normalize_channel(img_channel_i, lower=1, upper=99)
|
39 |
+
|
40 |
my_model = MultiStreamCellSegModel.from_pretrained("Lewislou/cellseg_sribd")
|
41 |
checkpoints = torch.load('model.pt')
|
42 |
my_model.__init__(ModelConfig())
|