Linaqruf commited on
Commit
951cf8d
·
verified ·
1 Parent(s): c438f1b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -5
app.py CHANGED
@@ -80,8 +80,8 @@ class MigrationTool:
80
  Tuple of (success, message)
81
  """
82
  try:
83
- # Create HubApi instance with token
84
- api = HubApi(token=token)
85
 
86
  # Determine visibility
87
  vis = ModelVisibility.PUBLIC if visibility == "public" else ModelVisibility.PRIVATE
@@ -102,7 +102,8 @@ class MigrationTool:
102
  model_id=repo_id,
103
  visibility=vis,
104
  license=lic,
105
- chinese_name=chinese_name
 
106
  )
107
  except Exception as create_error:
108
  # Repository might already exist, continue to push
@@ -112,13 +113,15 @@ class MigrationTool:
112
  if repo_type == "model":
113
  api.push_model(
114
  model_id=repo_id,
115
- model_dir=local_path
 
116
  )
117
  else:
118
  # For datasets, use similar approach
119
  api.push_model(
120
  model_id=repo_id,
121
- model_dir=local_path
 
122
  )
123
 
124
  return True, f"✓ Successfully uploaded {repo_type} to ModelScope"
 
80
  Tuple of (success, message)
81
  """
82
  try:
83
+ # Create HubApi instance
84
+ api = HubApi()
85
 
86
  # Determine visibility
87
  vis = ModelVisibility.PUBLIC if visibility == "public" else ModelVisibility.PRIVATE
 
102
  model_id=repo_id,
103
  visibility=vis,
104
  license=lic,
105
+ chinese_name=chinese_name,
106
+ token=token,
107
  )
108
  except Exception as create_error:
109
  # Repository might already exist, continue to push
 
113
  if repo_type == "model":
114
  api.push_model(
115
  model_id=repo_id,
116
+ model_dir=local_path,
117
+ token=token,
118
  )
119
  else:
120
  # For datasets, use similar approach
121
  api.push_model(
122
  model_id=repo_id,
123
+ model_dir=local_path,
124
+ token=token,
125
  )
126
 
127
  return True, f"✓ Successfully uploaded {repo_type} to ModelScope"