Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -80,8 +80,8 @@ class MigrationTool:
|
|
| 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,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"
|