EveSa commited on
Commit
8dba466
1 Parent(s): 1114f48

fix pre-commit and lsf

Browse files
Files changed (2) hide show
  1. src/api.py +10 -4
  2. src/model.py +1 -0
src/api.py CHANGED
@@ -10,6 +10,8 @@ from inference import inferenceAPI
10
  # appel de la fonction inference, adaptee pour une entree txt
11
  def summarize(text: str):
12
  return " ".join(inferenceAPI(text))
 
 
13
  # ----------------------------------------------------------------------------------
14
 
15
 
@@ -20,23 +22,27 @@ app = FastAPI()
20
  templates = Jinja2Templates(directory="templates")
21
  app.mount("/templates", StaticFiles(directory="templates"), name="templates")
22
 
 
23
  @app.get("/")
24
  async def index(request: Request):
25
  return templates.TemplateResponse("index.html.jinja", {"request": request})
26
 
 
27
  # retourner le texte, les predictions et message d'erreur si formulaire envoye vide
28
  @app.post("/")
29
  async def prediction(request: Request, text: str = Form(None)):
30
- if not text :
31
  error = "Merci de saisir votre texte."
32
  return templates.TemplateResponse(
33
- "index.html.jinja", {"request": request, "text": error}
34
- )
35
- else :
36
  summary = summarize(text)
37
  return templates.TemplateResponse(
38
  "index.html.jinja", {"request": request, "text": text, "summary": summary}
39
  )
 
 
40
  # ------------------------------------------------------------------------------------
41
 
42
 
 
10
  # appel de la fonction inference, adaptee pour une entree txt
11
  def summarize(text: str):
12
  return " ".join(inferenceAPI(text))
13
+
14
+
15
  # ----------------------------------------------------------------------------------
16
 
17
 
 
22
  templates = Jinja2Templates(directory="templates")
23
  app.mount("/templates", StaticFiles(directory="templates"), name="templates")
24
 
25
+
26
  @app.get("/")
27
  async def index(request: Request):
28
  return templates.TemplateResponse("index.html.jinja", {"request": request})
29
 
30
+
31
  # retourner le texte, les predictions et message d'erreur si formulaire envoye vide
32
  @app.post("/")
33
  async def prediction(request: Request, text: str = Form(None)):
34
+ if not text:
35
  error = "Merci de saisir votre texte."
36
  return templates.TemplateResponse(
37
+ "index.html.jinja", {"request": request, "text": error}
38
+ )
39
+ else:
40
  summary = summarize(text)
41
  return templates.TemplateResponse(
42
  "index.html.jinja", {"request": request, "text": text, "summary": summary}
43
  )
44
+
45
+
46
  # ------------------------------------------------------------------------------------
47
 
48
 
src/model.py CHANGED
@@ -8,6 +8,7 @@ import torch
8
 
9
  logging.basicConfig(level=logging.DEBUG)
10
 
 
11
  class Encoder(torch.nn.Module):
12
  def __init__(
13
  self,
 
8
 
9
  logging.basicConfig(level=logging.DEBUG)
10
 
11
+
12
  class Encoder(torch.nn.Module):
13
  def __init__(
14
  self,