Update app.py

app.py

@@ -131,20 +131,20 @@ class Bergamot:
             response, message_text = str(downloaderror), f"Error downloading {self.model_name}: {downloaderror}."
             print(downloaderror)
     def translate(self):
-    try:
-        model = self.service.modelFromConfigPath(f"{self.localfolder}/config.yml")
-        # model = self.service.modelFromConfig(localfolder)
-        options = bergamot.ResponseOptions(alignment=False, sentenceMappings=False, qualityScores=False, HTML=False)
-        rawresponse = self.service.translate(model, bergamot.VectorString(self.input_text), options)
-        response: list|str = [r.target.text for r in rawresponse] if len(rawresponse) > 1 else next(iter(rawresponse)).target.text
-        print(type(self.input_text), len(self.input_text), len(rawresponse), type(response), response)
-        # response = [r.target.text for r in model_response][0] if isinstance(response, bergamot._bergamot.VectorResponse) else next(iter(response)).target.text
-        # response is of type bergamot._bergamot.VectorResponse, an iterable of bergamot._bergamot.Response
-        message_text = f"Translated from {self.sl} to {self.tl} with {self.model_name}."
-    except Exception as translateerror:
-        response, message_text = str(translateerror), f"Error translating from {self.sl} to {self.tl} with {self.model_name}: {translateerror}."
-        print(translateerror)
-    return {"input": self.input_text, "translated_text": response, "message_text": message_text}
+        try:
+            model = self.service.modelFromConfigPath(f"{self.localfolder}/config.yml")
+            # model = self.service.modelFromConfig(localfolder)
+            options = bergamot.ResponseOptions(alignment=False, sentenceMappings=False, qualityScores=False, HTML=False)
+            rawresponse = self.service.translate(model, bergamot.VectorString(self.input_text), options)
+            response: list|str = [r.target.text for r in rawresponse] if len(rawresponse) > 1 else next(iter(rawresponse)).target.text
+            print(type(self.input_text), len(self.input_text), len(rawresponse), type(response), response)
+            # response = [r.target.text for r in model_response][0] if isinstance(response, bergamot._bergamot.VectorResponse) else next(iter(response)).target.text
+            # response is of type bergamot._bergamot.VectorResponse, an iterable of bergamot._bergamot.Response
+            message_text = f"Translated from {self.sl} to {self.tl} with {self.model_name}."
+        except Exception as translateerror:
+            response, message_text = str(translateerror), f"Error translating from {self.sl} to {self.tl} with {self.model_name}: {translateerror}."
+            print(translateerror)
+        return {"input": self.input_text, "translated_text": response, "message_text": message_text}
 
 # 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
 @app.get("/bergamot", operation_id="get_bergamot", description="Translate text with Bergamot", tags=["bergamot"], summary="Translate text with Bergamot")