Update app.py

app.py

@@ -60,9 +60,8 @@ class model:
 
         nucleus_image = processed_images[0].unsqueeze(0)
         protein_image = processed_images[1].unsqueeze(0)
-        protein_image = protein_image > 0
-        protein_image = 1.0 * protein_image
-                
+        protein_image = protein_image/torch.max(protein_image)
+        protein_image = 1 - protein_image                
 
         formatted_predicted_sequence = run_sequence_prediction(
             sequence_input=sequence_input,
@@ -110,7 +109,6 @@ with gr.Blocks(theme='gradio/soft') as demo:
         nucleus_image = gr.Image(
             source="upload", 
             tool="sketch",
-            invert_colors=True,
             label="Nucleus Image", 
             interactive=True,
             image_mode="L",