Merge pull request #3 from EveSa/Eve

Dockerfile

@@ -8,4 +8,8 @@ RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
 COPY . .
 
-CMD ["uvicorn", "api:app", "--host", "0.0.0.0", "--port", "7860"]
+EXPOSE 3001
+
+#CMD ["uvicorn", "--app-dir", "./src", "api:app", "--host", "127.0.0.1", "--port", "8888"]
+
+CMD python3 -m uvicorn --app-dir ./src api:app --host 0.0.0.0 --port 3001

requirements.txt

@@ -6,7 +6,6 @@ idna==3.4
 Jinja2==3.1.2
 joblib==1.2.0
 MarkupSafe==2.1.2
-nltk==3.8.1
 numpy==1.24.2
 nvidia-cublas-cu11==11.10.3.66
 nvidia-cuda-nvrtc-cu11==11.7.99

src/api.py

@@ -10,6 +10,8 @@ from inference import inferenceAPI
 # appel de la fonction inference, adaptee pour une entree txt
 def summarize(text: str):
     return " ".join(inferenceAPI(text))
+
+
 # ----------------------------------------------------------------------------------
 
 
@@ -20,23 +22,27 @@ app = FastAPI()
 templates = Jinja2Templates(directory="templates")
 app.mount("/templates", StaticFiles(directory="templates"), name="templates")
 
+
 @app.get("/")
 async def index(request: Request):
     return templates.TemplateResponse("index.html.jinja", {"request": request})
 
+
 # retourner le texte, les predictions et message d'erreur si formulaire envoye vide
 @app.post("/")
 async def prediction(request: Request, text: str = Form(None)):
-    if not text :
+    if not text:
         error = "Merci de saisir votre texte."
         return templates.TemplateResponse(
-        "index.html.jinja", {"request": request, "text": error}
-    )
-    else :
+            "index.html.jinja", {"request": request, "text": error}
+        )
+    else:
         summary = summarize(text)
         return templates.TemplateResponse(
             "index.html.jinja", {"request": request, "text": text, "summary": summary}
         )
+
+
 # ------------------------------------------------------------------------------------
 
 

src/model.py

@@ -8,6 +8,7 @@ import torch
 
 logging.basicConfig(level=logging.DEBUG)
 
+
 class Encoder(torch.nn.Module):
     def __init__(
         self,