Merge pull request #12 from EveSa/EveBis

fix(function): fix name function and their calls

src/api.py

@@ -2,17 +2,17 @@ from fastapi import FastAPI, Form, Request
 from fastapi.staticfiles import StaticFiles
 from fastapi.templating import Jinja2Templates
 
-from src.inference_lstm import inferenceAPI
-from src.inference_t5 import inferenceAPI
+from src.inference_lstm import inference_lstm
+from src.inference_t5 import inference_t5
 
 
 # ------ INFERENCE MODEL --------------------------------------------------------------
 # appel de la fonction inference, adaptee pour une entree txt
 def summarize(text: str):
     if choisir_modele.var == "lstm":
-        return " ".join(inferenceAPI(text))
+        return " ".join(inference_lstm(text))
     elif choisir_modele.var == "fineTunedT5":
-        text = inferenceAPI_t5(text)
+        text = inference_t5(text)
 
 
 # ----------------------------------------------------------------------------------

src/inference_lstm.py

@@ -15,7 +15,7 @@ with open("model/vocab.pkl", "rb") as vocab:
 vectoriser = dataloader.Vectoriser(words)
 
 
-def inferenceAPI(text: str) -> str:
+def inference_lstm(text: str) -> str:
     """
     Predict the summary for an input text
     --------

src/inference_t5.py

@@ -17,7 +17,7 @@ def clean_text(texts: str) -> str:
     return texts
 
 
-def inferenceAPI(text: str) -> str:
+def inference_t5(text: str) -> str:
     """
     Predict the summary for an input text
     --------