pyproject.toml

@@ -16,9 +16,6 @@ uvicorn = "^0.14.0"
 isort = "^5.8.0"
 pyflakes = "^2.3.1"
 
-[tool.poetry.scripts]
-api = "locimend.api:main"
-
 [build-system]
 requires = ["poetry-core>=1.0.0"]
 build-backend = "poetry.core.masonry.api"

src/api.py

@@ -1,8 +1,7 @@
 from fastapi import FastAPI
 from pydantic import BaseModel
-from uvicorn import run
 
-from locimend.model import infer_sequence
+from model import infer_sequence
 
 app = FastAPI()
 
@@ -21,7 +20,3 @@ async def get_sequence_path(sequence: str):
 async def get_sequence_body(sequence: Input):
     correct_sequence = await infer_sequence(sequence.sequence)
     return {"sequence": correct_sequence}
-
-
-def main():
-    run(app, host="0.0.0.0")

src/model.py

@@ -10,13 +10,8 @@ from tensorflow.keras.optimizers import Adam
 from tensorflow.keras.regularizers import l2
 from tensorflow.random import set_seed
 
-from locimend.hyperparameters import Hyperparameters
-from locimend.preprocessing import (
-    BASES,
-    dataset_creation,
-    decode_sequence,
-    encode_sequence,
-)
+from hyperparameters import Hyperparameters
+from preprocessing import BASES, dataset_creation, decode_sequence, encode_sequence
 
 
 def build_model(hyperparams) -> Model: