strangekitten commited on
Commit
88eb143
1 Parent(s): 89899dd

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -1,7 +1,7 @@
1
  import torch
2
  import numpy as np
3
 
4
- from .classes_desc import classes, classes_desc
5
 
6
 
7
  def get_most_probability_terms(logits, top_k=5):
 
1
  import torch
2
  import numpy as np
3
 
4
+ from classes_desc import classes, classes_desc
5
 
6
 
7
  def get_most_probability_terms(logits, top_k=5):