Spaces:
Runtime error
Runtime error
debug api
Browse files- src/api.py +2 -2
- src/inference_lstm.py +2 -2
src/api.py
CHANGED
@@ -2,8 +2,8 @@ from fastapi import FastAPI, Form, Request
|
|
2 |
from fastapi.staticfiles import StaticFiles
|
3 |
from fastapi.templating import Jinja2Templates
|
4 |
|
5 |
-
from inference_lstm import inferenceAPI
|
6 |
-
from inference_t5 import inferenceAPI
|
7 |
|
8 |
|
9 |
# ------ INFERENCE MODEL --------------------------------------------------------------
|
|
|
2 |
from fastapi.staticfiles import StaticFiles
|
3 |
from fastapi.templating import Jinja2Templates
|
4 |
|
5 |
+
from src.inference_lstm import inferenceAPI
|
6 |
+
from src.inference_t5 import inferenceAPI
|
7 |
|
8 |
|
9 |
# ------ INFERENCE MODEL --------------------------------------------------------------
|
src/inference_lstm.py
CHANGED
@@ -6,8 +6,8 @@ import pickle
|
|
6 |
|
7 |
import torch
|
8 |
|
9 |
-
import dataloader
|
10 |
-
from model import Decoder, Encoder, EncoderDecoderModel
|
11 |
# from transformers import AutoModel
|
12 |
|
13 |
with open("model/vocab.pkl", "rb") as vocab:
|
|
|
6 |
|
7 |
import torch
|
8 |
|
9 |
+
from src import dataloader
|
10 |
+
from src.model import Decoder, Encoder, EncoderDecoderModel
|
11 |
# from transformers import AutoModel
|
12 |
|
13 |
with open("model/vocab.pkl", "rb") as vocab:
|