TiberiuCristianLeon commited on
Commit
d5c671b
·
verified ·
1 Parent(s): 8b204d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -65,13 +65,13 @@ def bergamot(input_text: list[str] = Query(description="Input list of strings"),
65
  models = service.modelFromConfigPath("./deen/bergamot.config.yml")
66
  options = bergamot.ResponseOptions(alignment=False, qualityScores=False, HTML=False)
67
  rawresponse = service.translate(models, bergamot.VectorString(input_text), options)
68
- response = [r.target.text for r in rawresponse] if len(rawresponse) > 1 else next(iter(rawresponse)).target.text
69
- print(type(input_text), len(input_text), len(rawresponse), response)
70
  # response = [r.target.text for r in model_response][0] if isinstance(response, bergamot._bergamot.VectorResponse) else next(iter(response)).target.text
71
  # response is of type bergamot._bergamot.VectorResponse, an iterable of bergamot._bergamot.Response
72
  except Exception as error:
73
  response = error
74
- return {"input": input_text, "result": response, "model": str(model)}
75
 
76
  # Create an MCP server based on this app
77
  mcp = FastApiMCP(
 
65
  models = service.modelFromConfigPath("./deen/bergamot.config.yml")
66
  options = bergamot.ResponseOptions(alignment=False, qualityScores=False, HTML=False)
67
  rawresponse = service.translate(models, bergamot.VectorString(input_text), options)
68
+ response: list|str = [r.target.text for r in rawresponse] if len(rawresponse) > 1 else next(iter(rawresponse)).target.text
69
+ print(type(input_text), len(input_text), len(rawresponse), type(response), response)
70
  # response = [r.target.text for r in model_response][0] if isinstance(response, bergamot._bergamot.VectorResponse) else next(iter(response)).target.text
71
  # response is of type bergamot._bergamot.VectorResponse, an iterable of bergamot._bergamot.Response
72
  except Exception as error:
73
  response = error
74
+ return {"input": input_text, "result": response, "model": model}
75
 
76
  # Create an MCP server based on this app
77
  mcp = FastApiMCP(