Update app.py

app.py

@@ -53,9 +53,9 @@ class model:
         protein_image = image['mask'].convert('L')
 
         to_tensor = T.ToTensor()
-        nucleus_tensor = to_tensor(nucleus_image)
-        protein_tensor = to_tensor(protein_image)
-        stacked_images = torch.stack([nucleus_tensor, protein_tensor], dim=0)
+        nucleus_image = to_tensor(nucleus_image)
+        protein_image = to_tensor(protein_image)
+        stacked_images = torch.stack([nucleus_image, protein_image], dim=0)
         processed_images = process_image(stacked_images, dataset)
 
         nucleus_image = processed_images[0].unsqueeze(0)