Emaad commited on
Commit
5a8c0ad
1 Parent(s): eaf0b75

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -115,8 +115,7 @@ class model:
115
 
116
  nucleus_image = nucleus_image.unsqueeze(0)
117
  nucleus_image = process_image(nucleus_image, dataset, 'nucleus')
118
- nucleus_image = nucleus_image.unsqueeze(0)
119
- protein_image = protein_image.unsqueeze(0).unsqueeze(0)
120
  #protein_image = 1.0*(protein_image > .01)
121
 
122
 
 
115
 
116
  nucleus_image = nucleus_image.unsqueeze(0)
117
  nucleus_image = process_image(nucleus_image, dataset, 'nucleus')
118
+ protein_image = protein_image.unsqueeze(0)
 
119
  #protein_image = 1.0*(protein_image > .01)
120
 
121