TiberiuCristianLeon commited on
Commit
7db8a97
·
verified ·
1 Parent(s): 42c14b6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -44,8 +44,10 @@ def paraphrase(text: str, model: str):
44
 
45
  @app.get("/translate", operation_id="get_translate", description="Translate text", tags=["translate"], summary="Translate text")
46
  def translate(text: str, model: Optional[str] = MTMODELS['gemma']):
47
- # resultValue, exception = Translate.paraphraseTranslateMethod(text, model)
48
- resultValue: str = Translate.gemma_direct(text, model)
 
 
49
  return {"input": text, "result": resultValue, "model": model}
50
 
51
  # Create an MCP server based on this app
 
44
 
45
  @app.get("/translate", operation_id="get_translate", description="Translate text", tags=["translate"], summary="Translate text")
46
  def translate(text: str, model: Optional[str] = MTMODELS['gemma']):
47
+ if 'BlackKakapo' in model:
48
+ resultValue = Translate.paraphraseTranslateMethod(text, model)
49
+ else:
50
+ resultValue: str = Translate.gemma_direct(text, model)
51
  return {"input": text, "result": resultValue, "model": model}
52
 
53
  # Create an MCP server based on this app