Update app.py

app.py

@@ -90,7 +90,7 @@ def calculate_shap_values(model, x_tensor):
         # Baseline
         baseline_output = model(x_tensor)
         baseline_probs = torch.softmax(baseline_output, dim=1)
-        baseline_prob = baseline_probs[0, 1].item()  # Probability of 'human' class
+        baseline_prob = baseline_probs[0, 1].item()  # Probability of 'human'
         
         # Zeroing each feature to measure impact
         shap_values = []
@@ -100,7 +100,7 @@ def calculate_shap_values(model, x_tensor):
             x_zeroed[0, i] = 0.0
             output = model(x_zeroed)
             probs = torch.softmax(output, dim=1)
-            prob = probs[0, 1].item()
+            prob = probs[0, 1].item()  # Probability of 'human'
             impact = baseline_prob - prob
             shap_values.append(impact)
             x_zeroed[0, i] = original_val  # restore
@@ -354,6 +354,7 @@ def analyze_sequence(file_path, top_k=10, fasta_text="", window_size=500):
         with torch.no_grad():
             output = model(x_tensor)
             probs = torch.softmax(output, dim=1)
+            # Using index 1 for probability of human
             pred_human = probs[0, 1].item()
             
         # Calculate SHAP values
@@ -814,4 +815,4 @@ if __name__ == "__main__":
         server_port=7860,        # Default Gradio port
         show_api=False,          # Hide API docs
         debug=False              # Set to True for debugging
-    )
+    )