umair007 commited on
Commit
0a7c45e
·
1 Parent(s): 6d7041e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -10
app.py CHANGED
@@ -18,10 +18,10 @@ def download_file(file_url_1, file_name_1):
18
  gdown.download(url=file_url_1, output=f"file/{file_name_1}", quiet=False, fuzzy=True)
19
  return "download file done!"
20
  elif "civitai.com" in file_url_1:
21
- os.system(f"aria2c --dir=file/{file_name_1} --summary-interval=10 -c -x 10 -k 1M -s 10 {file_url_1}")
22
  return "download file done!"
23
  elif "huggingface.co" in file_url_1:
24
- os.system(f"aria2c --dir=file/{file_name_1} --summary-interval=10 -c -x 10 -k 1M -s 10 {file_url_1}")
25
  return "download file done!"
26
  else:
27
  try:
@@ -74,10 +74,10 @@ def download_ckpt(ckpt_url, ckpt_name):
74
  gdown.download(url=ckpt_url, output=f"ckpt/{ckpt_name}.ckpt", quiet=False, fuzzy=True)
75
  return "download ckpt done!"
76
  elif "civitai.com" in ckpt_url:
77
- os.system(f"aria2c --dir=ckpt/{ckpt_name} --summary-interval=10 -c -x 10 -k 1M -s 10 {ckpt_url}")
78
  return "download ckpt done!"
79
  elif "huggingface.co" in ckpt_url:
80
- os.system(f"aria2c --dir=ckpt/{ckpt_name} --summary-interval=10 -c -x 10 -k 1M -s 10 {ckpt_url}")
81
  return "download ckpt done!"
82
  else:
83
  try:
@@ -204,10 +204,10 @@ def download_ckpt(ckpt_url_2, ckpt_name_2):
204
  gdown.download(url=ckpt_url_2, output=f"ckpt/{ckpt_name_2}.ckpt", quiet=False, fuzzy=True)
205
  return "download ckpt done!"
206
  elif "civitai.com" in ckpt_url_2:
207
- os.system(f"aria2c --dir=ckpt/{ckpt_name_2}.ckpt --summary-interval=10 -c -x 10 -k 1M -s 10 {ckpt_url_2}")
208
  return "download ckpt done!"
209
  elif "huggingface.co" in ckpt_url_2:
210
- os.system(f"aria2c --dir=ckpt/{ckpt_name_2}.ckpt --summary-interval=10 -c -x 10 -k 1M -s 10 {ckpt_url_2}")
211
  return "download ckpt done!"
212
  else:
213
  try:
@@ -265,10 +265,10 @@ def download_safetensors(safetensors_url, safetensors_name):
265
  gdown.download(url=safetensors_url, output=f"safetensors/{safetensors_name}.safetensors", quiet=False, fuzzy=True)
266
  return "download safetensors done!"
267
  elif "civitai.com" in safetensors_url:
268
- os.system(f"aria2c --dir=safetensors/{safetensors_name}.safetensors --summary-interval=10 -c -x 10 -k 1M -s 10 {safetensors_url}")
269
  return "download safetensors done!"
270
  elif "huggingface.co" in safetensors_url:
271
- os.system(f"aria2c --dir=safetensors/{safetensors_name}.safetensors --summary-interval=10 -c -x 10 -k 1M -s 10 {safetensors_url}")
272
  return "download safetensors done!"
273
  else:
274
  try:
@@ -357,7 +357,7 @@ with block:
357
  """
358
  ### download and push file ckpt, yaml, safetensors etc <br />
359
  file_url = https://civitai.com/api/download/models/4224 or https://huggingface.co/prompthero/openjourney/resolve/main/mdjrny-v4.ckpt or https://drive.google.com/file/d/file-id/view?usp=share_link<br />
360
- file_name = openjourney <br />
361
  extension_name = ckpt, yaml, safetensors etc whatsever your file extension_name is <br />
362
  ckpt_model_to = camenduru/openjourney <br />
363
  branch = ckpt <br />
@@ -367,7 +367,7 @@ with block:
367
  with gr.Box():
368
  with gr.Row().style(equal_height=True):
369
  text_file_url_1 = gr.Textbox(show_label=False, max_lines=1, placeholder="file_url")
370
- text_file_name_1 = gr.Textbox(show_label=False, max_lines=1, placeholder="file_name")
371
  text_file_model_to = gr.Textbox(show_label=False, max_lines=1, placeholder="file_model_to")
372
  text_file_branch = gr.Textbox(show_label=False, value="file", max_lines=1, placeholder="file_branch")
373
  text_file_token = gr.Textbox(show_label=False, max_lines=1, placeholder="🤗 token")
 
18
  gdown.download(url=file_url_1, output=f"file/{file_name_1}", quiet=False, fuzzy=True)
19
  return "download file done!"
20
  elif "civitai.com" in file_url_1:
21
+ os.system(f"aria2c --out=file/{file_name_1} --summary-interval=10 -c -x 10 -k 1M -s 10 {file_url_1}")
22
  return "download file done!"
23
  elif "huggingface.co" in file_url_1:
24
+ os.system(f"aria2c --out=file/{file_name_1} --summary-interval=10 -c -x 10 -k 1M -s 10 {file_url_1}")
25
  return "download file done!"
26
  else:
27
  try:
 
74
  gdown.download(url=ckpt_url, output=f"ckpt/{ckpt_name}.ckpt", quiet=False, fuzzy=True)
75
  return "download ckpt done!"
76
  elif "civitai.com" in ckpt_url:
77
+ os.system(f"aria2c --out=ckpt/{ckpt_name} --summary-interval=10 -c -x 10 -k 1M -s 10 {ckpt_url}")
78
  return "download ckpt done!"
79
  elif "huggingface.co" in ckpt_url:
80
+ os.system(f"aria2c --out=ckpt/{ckpt_name} --summary-interval=10 -c -x 10 -k 1M -s 10 {ckpt_url}")
81
  return "download ckpt done!"
82
  else:
83
  try:
 
204
  gdown.download(url=ckpt_url_2, output=f"ckpt/{ckpt_name_2}.ckpt", quiet=False, fuzzy=True)
205
  return "download ckpt done!"
206
  elif "civitai.com" in ckpt_url_2:
207
+ os.system(f"aria2c --out=ckpt/{ckpt_name_2}.ckpt --summary-interval=10 -c -x 10 -k 1M -s 10 {ckpt_url_2}")
208
  return "download ckpt done!"
209
  elif "huggingface.co" in ckpt_url_2:
210
+ os.system(f"aria2c --out=ckpt/{ckpt_name_2}.ckpt --summary-interval=10 -c -x 10 -k 1M -s 10 {ckpt_url_2}")
211
  return "download ckpt done!"
212
  else:
213
  try:
 
265
  gdown.download(url=safetensors_url, output=f"safetensors/{safetensors_name}.safetensors", quiet=False, fuzzy=True)
266
  return "download safetensors done!"
267
  elif "civitai.com" in safetensors_url:
268
+ os.system(f"aria2c --out=safetensors/{safetensors_name}.safetensors --summary-interval=10 -c -x 10 -k 1M -s 10 {safetensors_url}")
269
  return "download safetensors done!"
270
  elif "huggingface.co" in safetensors_url:
271
+ os.system(f"aria2c --out=safetensors/{safetensors_name}.safetensors --summary-interval=10 -c -x 10 -k 1M -s 10 {safetensors_url}")
272
  return "download safetensors done!"
273
  else:
274
  try:
 
357
  """
358
  ### download and push file ckpt, yaml, safetensors etc <br />
359
  file_url = https://civitai.com/api/download/models/4224 or https://huggingface.co/prompthero/openjourney/resolve/main/mdjrny-v4.ckpt or https://drive.google.com/file/d/file-id/view?usp=share_link<br />
360
+ file_name = openjourney.ckpt or openjourney.safetensors whatsever your extension is <br />
361
  extension_name = ckpt, yaml, safetensors etc whatsever your file extension_name is <br />
362
  ckpt_model_to = camenduru/openjourney <br />
363
  branch = ckpt <br />
 
367
  with gr.Box():
368
  with gr.Row().style(equal_height=True):
369
  text_file_url_1 = gr.Textbox(show_label=False, max_lines=1, placeholder="file_url")
370
+ text_file_name_1 = gr.Textbox(show_label=False, max_lines=1, placeholder="file_name ex file.ckpt ")
371
  text_file_model_to = gr.Textbox(show_label=False, max_lines=1, placeholder="file_model_to")
372
  text_file_branch = gr.Textbox(show_label=False, value="file", max_lines=1, placeholder="file_branch")
373
  text_file_token = gr.Textbox(show_label=False, max_lines=1, placeholder="🤗 token")