Kseniia-Kholina commited on
Commit
71f3337
·
verified ·
1 Parent(s): 84bf6f4

Update heatmap.py

Browse files
Files changed (1) hide show
  1. heatmap.py +2 -0
heatmap.py CHANGED
@@ -5,6 +5,8 @@ import torch
5
  import matplotlib.pyplot as plt
6
  import seaborn as sns
7
  import numpy as np
 
 
8
 
9
 
10
  logging.getLogger("transformers.modeling_utils").setLevel(logging.ERROR)
 
5
  import matplotlib.pyplot as plt
6
  import seaborn as sns
7
  import numpy as np
8
+ import gradio as gr
9
+
10
 
11
 
12
  logging.getLogger("transformers.modeling_utils").setLevel(logging.ERROR)