Spaces:
Runtime error
Runtime error
EveSa
commited on
Commit
•
608047e
1
Parent(s):
0ed9611
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)
|