Spaces:
Running
Running
Commit
·
ebe5014
1
Parent(s):
b5ea46e
Revert "load app"
Browse filesThis reverts commit 953ee3625e7d670b4c0d3574223d7e0290d50295.
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 |
-
|
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):
|