Dusan commited on
Commit
4183d85
1 Parent(s): 0244cee

Update fudge/predict_clickbait.py

Browse files
Files changed (1) hide show
  1. fudge/predict_clickbait.py +4 -4
fudge/predict_clickbait.py CHANGED
@@ -15,10 +15,10 @@ import torch.nn.functional as F
15
  from transformers import AutoTokenizer, AutoModelWithLMHead
16
  from torch import Tensor
17
 
18
- from data import Dataset
19
- from model import Model
20
- from util import num_params
21
- from constants import *
22
 
23
 
24
 
 
15
  from transformers import AutoTokenizer, AutoModelWithLMHead
16
  from torch import Tensor
17
 
18
+ from fudge.data import Dataset
19
+ from fudge.model import Model
20
+ from fudge.util import num_params
21
+ from fudge.constants import *
22
 
23
 
24