4kasha commited on
Commit
e7088f8
1 Parent(s): 334e1b4

fix import

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import numpy as np
4
  import torch
5
  from transformers import AutoTokenizer, AutoModel
6
  from aligner import Aligner
7
- from utils import align_matrix_heatmap, plot_align_matrix_heatmap
8
 
9
  device = "cuda" if torch.cuda.is_available() else "cpu"
10
  torch.manual_seed(42)
 
4
  import torch
5
  from transformers import AutoTokenizer, AutoModel
6
  from aligner import Aligner
7
+ from utils import plot_align_matrix_heatmap
8
 
9
  device = "cuda" if torch.cuda.is_available() else "cpu"
10
  torch.manual_seed(42)