Update app.py

app.py

@@ -87,13 +87,14 @@ class model:
             model=self.model,
             device=device,
         )
-
+        
+        formatted_predicted_sequence = formatted_predicted_sequence[0]
         formatted_predicted_sequence = formatted_predicted_sequence.replace("<pad>","")
         formatted_predicted_sequence = formatted_predicted_sequence("<start>","")
         formatted_predicted_sequence = formatted_predicted_sequence("<end>","")
         formatted_predicted_sequence = bold_predicted_letters(sequence_input, formatted_predicted_sequence)                                                      
         
-        return T.ToPILImage()(protein_image[0,0]), T.ToPILImage()(nucleus_image[0,0]), 
+        return T.ToPILImage()(protein_image[0,0]), T.ToPILImage()(nucleus_image[0,0]), formatted_predicted_sequence
 
 base_class = model()