TiberiuCristianLeon commited on
Commit
e32216a
·
verified ·
1 Parent(s): e747345

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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}."