Update app.py
Browse files
app.py
CHANGED
@@ -62,7 +62,7 @@ def process_sequence(sequence, domain_bounds, n):
|
|
62 |
|
63 |
plt.figure(figsize=(15, 8))
|
64 |
sns.heatmap(transposed_logits_array, cmap='plasma', xticklabels=x_tick_labels, yticklabels=filtered_tokens)
|
65 |
-
plt.title('
|
66 |
plt.ylabel('Token')
|
67 |
plt.xlabel('Residue Index')
|
68 |
plt.yticks(rotation=0)
|
|
|
62 |
|
63 |
plt.figure(figsize=(15, 8))
|
64 |
sns.heatmap(transposed_logits_array, cmap='plasma', xticklabels=x_tick_labels, yticklabels=filtered_tokens)
|
65 |
+
plt.title('Token Probability Heatmap')
|
66 |
plt.ylabel('Token')
|
67 |
plt.xlabel('Residue Index')
|
68 |
plt.yticks(rotation=0)
|