Update main.py

main.py

@@ -40,28 +40,28 @@ async def ocr(
     return jsonable_encoder({"text": text})
 
 
-@app.post("/api/translate", response_model=dict)
-async def translate(
-    api_key: str = Depends(get_api_key),
-    text: str = Body(...),
-    src: str = "en",
-    trg: str = "zh",
-):
+#@app.post("/api/translate", response_model=dict)
+#async def translate(
+    #api_key: str = Depends(get_api_key),
+    #text: str = Body(...),
+    #src: str = "en",
+    #trg: str = "zh",
+#):
     # if api_key != API_KEY:
     #     return {"error": "Invalid API key"}, 401
 
-    tokenizer, model = get_model(src, trg)
+   # tokenizer, model = get_model(src, trg)
 
-    translated_text = ""
-    for sentence in sent_tokenize(text):
-        translated_sub = model.generate(**tokenizer(sentence, return_tensors="pt"))[0]
-        translated_text += tokenizer.decode(translated_sub, skip_special_tokens=True) + "\n"
+   # translated_text = ""
+   # for sentence in sent_tokenize(text):
+    #    translated_sub = model.generate(**tokenizer(sentence, return_tensors="pt"))[0]
+     #   translated_text += tokenizer.decode(translated_sub, skip_special_tokens=True) + "\n"
 
-    return jsonable_encoder({"translated_text": translated_text})
+   # return jsonable_encoder({"translated_text": translated_text})
 
 
-def get_model(src: str, trg: str):
-    model_name = f"Helsinki-NLP/opus-mt-{src}-{trg}"
-    tokenizer = MarianTokenizer.from_pretrained(model_name)
-    model = MarianMTModel.from_pretrained(model_name)
-    return tokenizer, model
+#def get_model(src: str, trg: str):
+   # model_name = f"Helsinki-NLP/opus-mt-{src}-{trg}"
+    #tokenizer = MarianTokenizer.from_pretrained(model_name)
+    #model = MarianMTModel.from_pretrained(model_name)
+    #return tokenizer, model