Spaces:
Running
Running
Update app.py
Browse files
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 |
-
|
| 132 |
-
|
| 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")
|