Spaces:
Runtime error
Runtime error
Update worker_runpod.py
Browse files- 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="
|
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)
|