Update app.py

app.py

@@ -62,9 +62,7 @@ class model:
  protein_image = processed_images[1].unsqueeze(0)
  protein_image = protein_image > 0
  protein_image = 1.0 * protein_image
- 
- print(f'{protein_image.sum()}')
- 
+ 
 
  formatted_predicted_sequence = run_sequence_prediction(
  sequence_input=sequence_input,
@@ -74,7 +72,7 @@ class model:
  device=device,
  )
  
- return T.ToPILImage()(protein_image), T.ToPILImage()(nucleus_image), formatted_predicted_sequence
+ return T.ToPILImage()(protein_image[0,0]), T.ToPILImage()(nucleus_image[0,0]), formatted_predicted_sequence
 
 base_class = model()