Lewislou commited on
Commit
24917f4
·
1 Parent(s): a41d6d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- pre_img_data[:,:,i] = normalize_channel(img_channel_i, lower=1, upper=99)
39
- #dummy_input = np.zeros((512,512,3)).astype(np.uint8)
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())