kovacsvi commited on
Commit
04f8f8b
·
1 Parent(s): 70be539

missing torch import

Browse files
Files changed (1) hide show
  1. interfaces/cap_minor_media.py +1 -0
interfaces/cap_minor_media.py CHANGED
@@ -6,6 +6,7 @@ import numpy as np
6
  import pandas as pd
7
  from transformers import AutoModelForSequenceClassification
8
  from transformers import AutoTokenizer
 
9
  from huggingface_hub import HfApi
10
  from collections import defaultdict
11
 
 
6
  import pandas as pd
7
  from transformers import AutoModelForSequenceClassification
8
  from transformers import AutoTokenizer
9
+ import torch.nn.functional as F
10
  from huggingface_hub import HfApi
11
  from collections import defaultdict
12