Wauplin HF staff commited on
Commit
6cda51e
1 Parent(s): db13d1d
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -46,15 +46,18 @@ you've downloaded from your Kaggle user account.
46
 
47
 
48
  def import_model(kaggle_model: str, repo_name: str, token: str) -> Iterable[List[Log]]:
 
 
49
  if not kaggle_model:
50
- return "Kaggle model is required."
 
51
  if not repo_name:
52
  repo_name = kaggle_model.split("/")[-1]
53
  if not token:
54
- return "HF Write Token is required."
 
55
  api = HfApi(token=token)
56
 
57
- runner = LogsViewRunner()
58
 
59
  yield runner.log(f"Creating HF repo {repo_name}")
60
  repo_url = api.create_repo(repo_name, exist_ok=True)
 
46
 
47
 
48
  def import_model(kaggle_model: str, repo_name: str, token: str) -> Iterable[List[Log]]:
49
+ runner = LogsViewRunner()
50
+
51
  if not kaggle_model:
52
+ yield runner.log("Kaggle model is required.", level="ERROR")
53
+ raise gr.Error("Kaggle model is required.")
54
  if not repo_name:
55
  repo_name = kaggle_model.split("/")[-1]
56
  if not token:
57
+ yield runner.log("HF Write Token is required.", level="ERROR")
58
+ raise gr.Error("HF Write Token is required.")
59
  api = HfApi(token=token)
60
 
 
61
 
62
  yield runner.log(f"Creating HF repo {repo_name}")
63
  repo_url = api.create_repo(repo_name, exist_ok=True)