debug api

src/api.py

@@ -2,8 +2,8 @@ from fastapi import FastAPI, Form, Request
 from fastapi.staticfiles import StaticFiles
 from fastapi.templating import Jinja2Templates
 
-from inference_lstm import inferenceAPI
-from inference_t5 import inferenceAPI
+from src.inference_lstm import inferenceAPI
+from src.inference_t5 import inferenceAPI
 
 
 # ------ INFERENCE MODEL --------------------------------------------------------------

src/inference_lstm.py

@@ -6,8 +6,8 @@ import pickle
 
 import torch
 
-import dataloader
-from model import Decoder, Encoder, EncoderDecoderModel
+from src import dataloader
+from src.model import Decoder, Encoder, EncoderDecoderModel
 # from transformers import AutoModel
 
 with open("model/vocab.pkl", "rb") as vocab: