Spaces:
Runtime error
Runtime error
EveSa
commited on
Update inference.py
Browse files- src/inference.py +2 -2
src/inference.py
CHANGED
@@ -5,8 +5,8 @@ import pickle
|
|
5 |
|
6 |
import torch
|
7 |
|
8 |
-
import dataloader
|
9 |
-
from model import Decoder, Encoder, EncoderDecoderModel
|
10 |
|
11 |
with open("model/vocab.pkl", "rb") as vocab:
|
12 |
words = pickle.load(vocab)
|
|
|
5 |
|
6 |
import torch
|
7 |
|
8 |
+
from src import dataloader
|
9 |
+
from src.model import Decoder, Encoder, EncoderDecoderModel
|
10 |
|
11 |
with open("model/vocab.pkl", "rb") as vocab:
|
12 |
words = pickle.load(vocab)
|