Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -94,7 +94,7 @@ class Bergamot:
|
|
| 94 |
self.repo_id = "TiberiuCristianLeon/Bergamot"
|
| 95 |
self.branches = ['base', 'base-memory', 'tiny']
|
| 96 |
self.subfolder = f"{sl}{tl}"
|
| 97 |
-
self.localfolder = f"{subfolder}/{model_name}"
|
| 98 |
# List all files in the repo
|
| 99 |
self.all_files = list_repo_files(repo_id, repo_type='model')
|
| 100 |
|
|
@@ -131,6 +131,7 @@ class Bergamot:
|
|
| 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)
|
|
@@ -168,7 +169,6 @@ def bergamot(input_text: list[str] = Query(description="Input string or list of
|
|
| 168 |
"""
|
| 169 |
try:
|
| 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}."
|
|
|
|
| 94 |
self.repo_id = "TiberiuCristianLeon/Bergamot"
|
| 95 |
self.branches = ['base', 'base-memory', 'tiny']
|
| 96 |
self.subfolder = f"{sl}{tl}"
|
| 97 |
+
self.localfolder = f"{self.subfolder}/{model_name}"
|
| 98 |
# List all files in the repo
|
| 99 |
self.all_files = list_repo_files(repo_id, repo_type='model')
|
| 100 |
|
|
|
|
| 131 |
response, message_text = str(downloaderror), f"Error downloading {self.model_name}: {downloaderror}."
|
| 132 |
print(downloaderror)
|
| 133 |
def translate(self):
|
| 134 |
+
self.downloadbergamotfiles()
|
| 135 |
try:
|
| 136 |
model = self.service.modelFromConfigPath(f"{self.localfolder}/config.yml")
|
| 137 |
# model = self.service.modelFromConfig(localfolder)
|
|
|
|
| 169 |
"""
|
| 170 |
try:
|
| 171 |
bergamotinstance = Bergamot(input_text, sl, tl, model_name)
|
|
|
|
| 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}."
|