Hjgugugjhuhjggg commited on
Commit
9c4d42f
·
verified ·
1 Parent(s): 606b557

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -59,7 +59,7 @@ class S3ModelLoader:
59
  )
60
 
61
  def _get_s3_uri(self, model_name):
62
- return f"s3://{self.bucket_name}/{model_name.replace('/', '-')}"
63
 
64
  def load_model_and_tokenizer(self, model_name):
65
  if model_name in token_dict:
 
59
  )
60
 
61
  def _get_s3_uri(self, model_name):
62
+ return f"s3://{self.bucket_name}/{model_name}"
63
 
64
  def load_model_and_tokenizer(self, model_name):
65
  if model_name in token_dict: