AnnasBlackHat commited on
Commit
ab7a2d4
1 Parent(s): a50cd18

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -18,10 +18,11 @@ except: print('images dir might already exist...')
18
  def download_local(url):
19
  resp = requests.get(url, allow_redirects=True)
20
  filename, file_extension = os.path.splitext(url)
 
21
  if file_extension == '':
22
- file_extension = '.jpg'
23
 
24
- filename = 'images/{}_{}{}'.format(int(time.time()), filename, file_extension)
25
  with open(filename, 'wb') as handler:
26
  handler.write(resp.content)
27
  return filename
 
18
  def download_local(url):
19
  resp = requests.get(url, allow_redirects=True)
20
  filename, file_extension = os.path.splitext(url)
21
+ filename = os.path.basename(url)
22
  if file_extension == '':
23
+ filename = filename+'.jpg'
24
 
25
+ filename = 'images/{}_{}'.format(int(time.time()), filename)
26
  with open(filename, 'wb') as handler:
27
  handler.write(resp.content)
28
  return filename