sreepathi-ravikumar commited on
Commit
3ca8773
·
verified ·
1 Parent(s): 271de0d

Update image_fetcher.py

Browse files
Files changed (1) hide show
  1. image_fetcher.py +8 -2
image_fetcher.py CHANGED
@@ -8,8 +8,14 @@ from io import BytesIO
8
  from duckduckgo_search import DDGS
9
 
10
  # Create folders
11
- os.makedirs("images", exist_ok=True)
12
- os.makedirs("images/tmp", exist_ok=True)
 
 
 
 
 
 
13
 
14
  # Helper: Random delay + headers
15
  def get_headers():
 
8
  from duckduckgo_search import DDGS
9
 
10
  # Create folders
11
+ DIRS = {
12
+ "base": "images",
13
+ "temp": os.path.join("images", "tmp")
14
+ }
15
+
16
+ # Ensure directories exist
17
+ for path in DIRS.values():
18
+ os.makedirs(path, exist_ok=True)
19
 
20
  # Helper: Random delay + headers
21
  def get_headers():