mulyavinaa commited on
Commit
c0f8474
1 Parent(s): 82d4d7d

Корректировка API, фикс переводчика

Browse files
Files changed (3) hide show
  1. api.py +5 -6
  2. mulyavin_aa/model/translator.py +4 -2
  3. 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.TranslatorRequest) \
40
- -> mulyavin_aa.model.translator.TranslatorResponse:
41
  """Перевод текста из Ru в En"""
42
- response = mulyavin_aa.model.translator.TranslatorResponse()
43
  pipe = mulyavin_aa.translator.load_text_translator_model()
44
- response.text = mulyavin_aa.translator.translate_to_en(request.text, pipe)
45
 
46
- return response
47
 
48
 
49
  # Запуск как приложения
50
  if __name__ == '__main__':
51
- uvicorn.run(app, port=8080, host='127.0.0.1')
 
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 TranslatorRequest(BaseModel):
 
5
  text: str
6
 
7
 
8
- class TranslatorResponse(BaseModel):
 
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