TiberiuCristianLeon commited on
Commit
2467a42
·
verified ·
1 Parent(s): 76e12db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -5
app.py CHANGED
@@ -128,14 +128,14 @@ class Bergamot:
128
  except Exception as dryrunerror:
129
  print('installed_pairs', self.installed_pairs, 'localfolder', self.localfolder, 'dry_runerror', dryrunerror)
130
  except Exception as downloaderror:
131
- response, message_text = str(downloaderror), f"Error downloading {self.model_name}: {downloaderror}."
132
- print(downloaderror)
133
  def translate(self):
134
  try:
135
- model = service.modelFromConfigPath(f"{self.localfolder}/config.yml")
136
- # model = service.modelFromConfig(localfolder)
137
  options = bergamot.ResponseOptions(alignment=False, sentenceMappings=False, qualityScores=False, HTML=False)
138
- rawresponse = service.translate(model, bergamot.VectorString(self.input_text), options)
139
  response: list|str = [r.target.text for r in rawresponse] if len(rawresponse) > 1 else next(iter(rawresponse)).target.text
140
  print(type(self.input_text), len(self.input_text), len(rawresponse), type(response), response)
141
  # response = [r.target.text for r in model_response][0] if isinstance(response, bergamot._bergamot.VectorResponse) else next(iter(response)).target.text
@@ -170,6 +170,9 @@ def bergamot(input_text: list[str] = Query(description="Input string or list of
170
  bergamotinstance = Bergamot(input_text, sl, tl, model_name)
171
  bergamotinstance.downloadbergamotfiles()
172
  return bergamotinstance.translate()
 
 
 
173
 
174
  # 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
175
  @app.get("/bergamots", operation_id="get_bergamot", description="Translate text with Bergamot", tags=["bergamot"], summary="Translate text with Bergamot")
 
128
  except Exception as dryrunerror:
129
  print('installed_pairs', self.installed_pairs, 'localfolder', self.localfolder, 'dry_runerror', dryrunerror)
130
  except Exception as downloaderror:
131
+ response, message_text = str(downloaderror), f"Error downloading {self.model_name}: {downloaderror}."
132
+ print(downloaderror)
133
  def translate(self):
134
  try:
135
+ model = self.service.modelFromConfigPath(f"{self.localfolder}/config.yml")
136
+ # model = self.service.modelFromConfig(localfolder)
137
  options = bergamot.ResponseOptions(alignment=False, sentenceMappings=False, qualityScores=False, HTML=False)
138
+ rawresponse = self.service.translate(model, bergamot.VectorString(self.input_text), options)
139
  response: list|str = [r.target.text for r in rawresponse] if len(rawresponse) > 1 else next(iter(rawresponse)).target.text
140
  print(type(self.input_text), len(self.input_text), len(rawresponse), type(response), response)
141
  # response = [r.target.text for r in model_response][0] if isinstance(response, bergamot._bergamot.VectorResponse) else next(iter(response)).target.text
 
170
  bergamotinstance = Bergamot(input_text, sl, tl, model_name)
171
  bergamotinstance.downloadbergamotfiles()
172
  return bergamotinstance.translate()
173
+ except Exception as mainerror:
174
+ response, message_text = str(mainerror), f"Error translating from {sl} to {tl} with {model_name}: {mainerror}."
175
+ print(mainerror)
176
 
177
  # 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
178
  @app.get("/bergamots", operation_id="get_bergamot", description="Translate text with Bergamot", tags=["bergamot"], summary="Translate text with Bergamot")