Spaces:

HuangLab
/

CELL-E_2-Image_Prediction

Sleeping

Emaad commited on May 17, 2023

Commit

b9ae49c

•

1 Parent(s): c40556d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -65,7 +65,7 @@ class model:
             protein_image = protein_image > 0
             protein_image = 1.0 * protein_image
         else:
-            nucleus_image = process_image(nucleus_image)
             protein_image = torch.ones((256, 256))
         threshold, heatmap = run_image_prediction(

             protein_image = protein_image > 0
             protein_image = 1.0 * protein_image
         else:
+            nucleus_image = process_image(nucleus_image).unsqueeze(0)
             protein_image = torch.ones((256, 256))
         threshold, heatmap = run_image_prediction(