gleisonnanet commited on
Commit
ebe5014
·
1 Parent(s): b5ea46e

Revert "load app"

Browse files

This reverts commit 953ee3625e7d670b4c0d3574223d7e0290d50295.

Files changed (1) hide show
  1. main.py +1 -2
main.py CHANGED
@@ -25,8 +25,7 @@ def load_model(pretrained_model: str = "facebook/m2m100_1.2B", cache_dir: str =
25
  model = M2M100ForConditionalGeneration.from_pretrained(pretrained_model, cache_dir=model_dir).to(device)
26
  model.eval()
27
  return tokenizer, model
28
- # aparentemente temos um problema ao carregar o modelo então vou tentar carregar no start da aplicação para não dar time-out na request
29
- load_model()
30
 
31
  @app.post("/translate")
32
  async def translate(request: TranslationRequest):
 
25
  model = M2M100ForConditionalGeneration.from_pretrained(pretrained_model, cache_dir=model_dir).to(device)
26
  model.eval()
27
  return tokenizer, model
28
+
 
29
 
30
  @app.post("/translate")
31
  async def translate(request: TranslationRequest):