TiberiuCristianLeon commited on
Commit
3dddf92
·
verified ·
1 Parent(s): b025c2f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -87,7 +87,7 @@ installed_pairs = set()
87
 
88
  # https://tiberiucristianleon-fastapimt.hf.space/bergamot?input_text=das%20ist%20keine%20gute%20Frau&input_text=das%20ist%20eine%20gute%20Nachricht&sl=de&tl=en&model=bergamot
89
  @app.get("/bergamot", operation_id="get_bergamot", description="Translate text with Bergamot", tags=["bergamot"], summary="Translate text with Bergamot")
90
- def bergamot(input_text: str = Query(description="Input strings"), sl: str = 'de', tl: str = 'en', model_name: Optional[str] = 'base/deen'):
91
  """
92
  Translates the input text from the source language to the target language using a specified model.
93
  Parameters:
@@ -132,12 +132,15 @@ def bergamot(input_text: str = Query(description="Input strings"), sl: str = 'de
132
  print(f"Downloaded to: {local_path}") # Downloaded to: deen/base/deen/config.yml
133
  # localfolder = local_path.rsplit('/', 1)[0]
134
  installed_pairs.add(subfolder)
135
- dry_run = hf_hub_download(repo_id=repo_id, subfolder=model_name, filename='config.yml', local_dir=subfolder)
136
- print('installed_pairs', installed_pairs, 'localfolder', localfolder, 'dry_run', dry_run)
 
 
 
137
  # model = service.modelFromConfigPath(f"{localfolder}/config.yml")
138
  model = service.modelFromConfigPath(localfolder)
139
  options = bergamot.ResponseOptions(alignment=False, qualityScores=False, HTML=False)
140
- rawresponse = service.translate(model, bergamot.VectorString([input_text]), options)
141
  response: list|str = [r.target.text for r in rawresponse] if len(rawresponse) > 1 else next(iter(rawresponse)).target.text
142
  print(type(input_text), len(input_text), len(rawresponse), type(response), response)
143
  # response = [r.target.text for r in model_response][0] if isinstance(response, bergamot._bergamot.VectorResponse) else next(iter(response)).target.text
 
87
 
88
  # https://tiberiucristianleon-fastapimt.hf.space/bergamot?input_text=das%20ist%20keine%20gute%20Frau&input_text=das%20ist%20eine%20gute%20Nachricht&sl=de&tl=en&model=bergamot
89
  @app.get("/bergamot", operation_id="get_bergamot", description="Translate text with Bergamot", tags=["bergamot"], summary="Translate text with Bergamot")
90
+ def bergamot(input_text: list[str] = Query(description="Input list of strings"), sl: str = 'de', tl: str = 'en', model_name: Optional[str] = 'base/deen'):
91
  """
92
  Translates the input text from the source language to the target language using a specified model.
93
  Parameters:
 
132
  print(f"Downloaded to: {local_path}") # Downloaded to: deen/base/deen/config.yml
133
  # localfolder = local_path.rsplit('/', 1)[0]
134
  installed_pairs.add(subfolder)
135
+ try:
136
+ dry_run = hf_hub_download(repo_id=repo_id, subfolder=model_name, filename='config.yml', local_dir=subfolder)
137
+ print('installed_pairs', installed_pairs, 'localfolder', localfolder, 'dry_run', dry_run)
138
+ except Exception as dryrunerror:
139
+ print('installed_pairs', installed_pairs, 'localfolder', localfolder, 'dry_run', dryrunerror)
140
  # model = service.modelFromConfigPath(f"{localfolder}/config.yml")
141
  model = service.modelFromConfigPath(localfolder)
142
  options = bergamot.ResponseOptions(alignment=False, qualityScores=False, HTML=False)
143
+ rawresponse = service.translate(model, bergamot.VectorString(input_text), options)
144
  response: list|str = [r.target.text for r in rawresponse] if len(rawresponse) > 1 else next(iter(rawresponse)).target.text
145
  print(type(input_text), len(input_text), len(rawresponse), type(response), response)
146
  # response = [r.target.text for r in model_response][0] if isinstance(response, bergamot._bergamot.VectorResponse) else next(iter(response)).target.text