Commit
·
c0f8474
1
Parent(s):
82d4d7d
Корректировка API, фикс переводчика
Browse files- api.py +5 -6
- mulyavin_aa/model/translator.py +4 -2
- requirements.txt +3 -0
api.py
CHANGED
@@ -36,16 +36,15 @@ def lang_detect(request: mulyavin_aa.model.langdetector.Request) \
|
|
36 |
|
37 |
|
38 |
@app.post("/translator/translate")
|
39 |
-
def lang_detect(request: mulyavin_aa.model.translator.
|
40 |
-
-> mulyavin_aa.model.translator.
|
41 |
"""Перевод текста из Ru в En"""
|
42 |
-
response = mulyavin_aa.model.translator.TranslatorResponse()
|
43 |
pipe = mulyavin_aa.translator.load_text_translator_model()
|
44 |
-
|
45 |
|
46 |
-
return
|
47 |
|
48 |
|
49 |
# Запуск как приложения
|
50 |
if __name__ == '__main__':
|
51 |
-
uvicorn.run(app, port=
|
|
|
36 |
|
37 |
|
38 |
@app.post("/translator/translate")
|
39 |
+
def lang_detect(request: mulyavin_aa.model.translator.Request) \
|
40 |
+
-> mulyavin_aa.model.translator.Response:
|
41 |
"""Перевод текста из Ru в En"""
|
|
|
42 |
pipe = mulyavin_aa.translator.load_text_translator_model()
|
43 |
+
text = mulyavin_aa.translator.translate_to_en(request.text, pipe)
|
44 |
|
45 |
+
return mulyavin_aa.model.translator.Response(text=text)
|
46 |
|
47 |
|
48 |
# Запуск как приложения
|
49 |
if __name__ == '__main__':
|
50 |
+
uvicorn.run(app, port=8000, host='127.0.0.1')
|
mulyavin_aa/model/translator.py
CHANGED
@@ -1,9 +1,11 @@
|
|
1 |
from pydantic import BaseModel
|
2 |
|
3 |
|
4 |
-
class
|
|
|
5 |
text: str
|
6 |
|
7 |
|
8 |
-
class
|
|
|
9 |
text: str
|
|
|
1 |
from pydantic import BaseModel
|
2 |
|
3 |
|
4 |
+
class Request(BaseModel):
|
5 |
+
"""Структура запроса"""
|
6 |
text: str
|
7 |
|
8 |
|
9 |
+
class Response(BaseModel):
|
10 |
+
"""Структура ответа"""
|
11 |
text: str
|
requirements.txt
CHANGED
@@ -8,3 +8,6 @@ sacremoses==0.1.1
|
|
8 |
translate==3.6.1
|
9 |
fastapi==0.104.1
|
10 |
uvicorn==0.24.0
|
|
|
|
|
|
|
|
8 |
translate==3.6.1
|
9 |
fastapi==0.104.1
|
10 |
uvicorn==0.24.0
|
11 |
+
|
12 |
+
pillow~=10.1.0
|
13 |
+
pydantic~=2.5.2
|