meepmoo commited on
Commit
17b5457
·
verified ·
1 Parent(s): 0933781

Update worker_runpod.py

Browse files
Files changed (1) hide show
  1. worker_runpod.py +1 -1
worker_runpod.py CHANGED
@@ -36,7 +36,7 @@ def to_pil(image):
36
  raise ValueError(f"Cannot convert {type(image)} to PIL.Image")
37
 
38
 
39
- def download_image(url, download_dir="asset"):
40
  # Ensure the download directory exists
41
  if not os.path.exists(download_dir):
42
  os.makedirs(download_dir, exist_ok=True)
 
36
  raise ValueError(f"Cannot convert {type(image)} to PIL.Image")
37
 
38
 
39
+ def download_image(url, download_dir="/content"):
40
  # Ensure the download directory exists
41
  if not os.path.exists(download_dir):
42
  os.makedirs(download_dir, exist_ok=True)