reinitialize importance scores

app.py

@@ -379,9 +379,6 @@ def generate_heatmap(protein_fasta):
         original_output = model(encoded_input["input_ids"], attention_mask=encoded_input["attention_mask"])
     original_probabilities = F.softmax(original_output.logits, dim=1).cpu().numpy()[0]
 
-    # Initialize an array to hold the importance scores
-    importance_scores = np.zeros((len(protein_sequence), len(original_probabilities)))
-
     # Define the size of each group
     group_size = 10  # You can change this
 
@@ -440,7 +437,7 @@ with gr.Blocks() as app:
     with gr.Tab("Single Sequence Prediction"):
         with gr.Row().style(equal_height=True):
             with gr.Column():
-                sequence = gr.inputs.Textbox(lines=15, placeholder='Enter Protein Sequence Here...', label="Protein Sequence")
+                sequence = gr.inputs.Textbox(lines=16, placeholder='Enter Protein Sequence Here...', label="Protein Sequence")
             with gr.Column():
                 with gr.Accordion("Example:"):
                     gr.Markdown("""