Update app.py

app.py

@@ -118,7 +118,7 @@ class model:
         protein_image = protein_image.unsqueeze(0)
         #protein_image = 1.0*(protein_image > .01)
         
-
+        print('test1')
         formatted_predicted_sequence = run_sequence_prediction(
             sequence_input=sequence_input,
             nucleus_image=nucleus_image,
@@ -126,7 +126,7 @@ class model:
             model=self.model,
             device=device,
         )
-        
+        print('test2')
         formatted_predicted_sequence = formatted_predicted_sequence[0]
         formatted_predicted_sequence = formatted_predicted_sequence.replace("<pad>","")
         formatted_predicted_sequence = formatted_predicted_sequence.replace("<cls>","")
@@ -180,7 +180,7 @@ with gr.Blocks(theme='gradio/soft') as demo:
         nucleus_image = gr.ImageMask(
             label = "Nucleus Image",
             interactive = "True",
-            #image_mode = "L",
+            image_mode = "L",
             brush_color = "#ffffff",
             type = "pil"
         )