TiberiuCristianLeon commited on
Commit
2ee9294
·
verified ·
1 Parent(s): d30b281

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -63,12 +63,12 @@ def listmodels():
63
  @app.get("/translate", operation_id="get_translate", description="Translate text", tags=["translate"], summary="Translate text")
64
  def translate(input_text: str, model_name: str = MODELS['mttcbig'], sl: str = 'en', tl: str = 'ro'):
65
  if 'BlackKakapo' in model_name:
66
- translation = src.Translate.paraphraseTranslateMethod(text, model_name)
67
  elif 'Helsinki-NLP' in model_name:
68
- translation = src.Translate.Translators.HelsinkiNLP_mulroa(model_name, sl, tl, input_text)
69
  else:
70
- translation: str = src.Translate.gemma_direct(text, model_name)
71
- return {"input": text, "result": translation, "model": model_name}
72
 
73
  # Keep track of installed (src, tgt) pairs
74
  installed_pairs = set()
 
63
  @app.get("/translate", operation_id="get_translate", description="Translate text", tags=["translate"], summary="Translate text")
64
  def translate(input_text: str, model_name: str = MODELS['mttcbig'], sl: str = 'en', tl: str = 'ro'):
65
  if 'BlackKakapo' in model_name:
66
+ translation = src.Translate.paraphraseTranslateMethod(input_text, model_name)
67
  elif 'Helsinki-NLP' in model_name:
68
+ translation = src.Translate.Translators(model_name, sl, tl, input_text).HelsinkiNLP_mulroa()
69
  else:
70
+ translation: str = src.Translate.gemma_direct(input_text, model_name)
71
+ return {"input": input_text, "result": translation, "model": model_name}
72
 
73
  # Keep track of installed (src, tgt) pairs
74
  installed_pairs = set()