anzorq commited on
Commit
b9fc382
1 Parent(s): de84b65

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -23,7 +23,7 @@ def get_ckpt_names(model_id = "nitrosocke/mo-di-diffusion"):
23
  if len(ckpt_files) == 0:
24
  return error_str("No checkpoint files found in the model repo."), None, None
25
 
26
- return None, gr.update(choices=ckpt_files, visible=True), gr.update(visible=True), "\n".join([f"- {ckpt_file.rfilename}" for ckpt_file in ckpt_files])
27
 
28
  except Exception as e:
29
  return error_str(e), None, None
@@ -31,7 +31,7 @@ def get_ckpt_names(model_id = "nitrosocke/mo-di-diffusion"):
31
  def convert(model_id, ckpt_name, token = "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"):
32
 
33
  model_id = url_to_model_id(model_id)
34
-
35
  # 1. Download the checkpoint file
36
  ckpt_path = hf_hub_download(repo_id=model_id, filename=ckpt_name)
37
 
 
23
  if len(ckpt_files) == 0:
24
  return error_str("No checkpoint files found in the model repo."), None, None
25
 
26
+ return None, gr.update(choices=ckpt_files, visible=True), gr.update(visible=True)
27
 
28
  except Exception as e:
29
  return error_str(e), None, None
 
31
  def convert(model_id, ckpt_name, token = "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"):
32
 
33
  model_id = url_to_model_id(model_id)
34
+
35
  # 1. Download the checkpoint file
36
  ckpt_path = hf_hub_download(repo_id=model_id, filename=ckpt_name)
37