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 |
-
|
| 9 |
-
from
|
| 10 |
|
| 11 |
with open("model/vocab.pkl", "rb") as vocab:
|
| 12 |
words = pickle.load(vocab)
|
|
|
|
| 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)
|