Debugging heatmap ticklabels in HF spaces

app.py

@@ -62,7 +62,8 @@ def create_scoring_matrix_visual(scores,sequence,AA_vocab=AA_vocab,mutation_rang
     heat = sns.heatmap(piv,cmap='RdYlGn',linewidths=0.30,ax=ax,vmin=np.percentile(scores.avg_score,2),vmax=np.percentile(scores.avg_score,98),\
                 cbar_kws={'label': 'Log likelihood ratio (mutant / starting sequence)'},annot_kws={"size": fontsize})
   heat.figure.axes[-1].yaxis.label.set_size(fontsize=int(fontsize*1.5))
-  #heat.figure.axes[-1].yaxis.set_ticklabels(heat.figure.axes[-1].yaxis.get_ticklabels(), fontsize=fontsize)
+  heat.figure.axes[-1].yaxis.set_ticklabels(heat.figure.axes[-1].yaxis.get_ticklabels(), fontsize=fontsize)
+  #heat.figure.axes[-1].yaxis.set_ticklabels(fontsize=fontsize)
   heat.set_title("Higher predicted scores (green) imply higher protein fitness",fontsize=fontsize*2, pad=40)
   heat.set_ylabel("Sequence position", fontsize = fontsize*2)
   heat.set_xlabel("Amino Acid mutation", fontsize = fontsize*2)

requirements.txt

@@ -1,6 +1,7 @@
 gradio
+torch
 transformers==4.17
 datasets==1.18.3
 biopython==1.78
-torch
-seaborn
+matplotlib==3.2.2
+seaborn==0.11.2