Frenchizer commited on
Commit
16892db
·
verified ·
1 Parent(s): 5384446

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -20,22 +20,22 @@ class TranslationInput(BaseModel):
20
  # FastAPI endpoint for model prediction
21
  @app.post("/predict")
22
  async def predict(translation_input: TranslationInput):
23
- """
24
  Endpoint for inference.
25
  :param translation_input: Text input in English.
26
  :return: Translated text in French.
27
  """
28
  # Tokenize input text
29
  tokenized_input = tokenizer(
30
- translation_input.input_text,
31
- return_tensors="np",
32
  padding=True
33
  )
34
  input_ids = tokenized_input["input_ids"]
35
 
36
  # Perform inference with ONNX model
37
  outputs = session.run(
38
- None,
39
  {"input_ids": input_ids.astype("int64")}
40
  )
41
 
 
20
  # FastAPI endpoint for model prediction
21
  @app.post("/predict")
22
  async def predict(translation_input: TranslationInput):
23
+ """
24
  Endpoint for inference.
25
  :param translation_input: Text input in English.
26
  :return: Translated text in French.
27
  """
28
  # Tokenize input text
29
  tokenized_input = tokenizer(
30
+ translation_input.input_text,
31
+ return_tensors="np",
32
  padding=True
33
  )
34
  input_ids = tokenized_input["input_ids"]
35
 
36
  # Perform inference with ONNX model
37
  outputs = session.run(
38
+ None,
39
  {"input_ids": input_ids.astype("int64")}
40
  )
41