Ahsen Khaliq commited on
Commit
8e018c1
1 Parent(s): 38a5c36

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -15,6 +15,8 @@ from stat import ST_CTIME
15
  from datetime import datetime, timedelta
16
  import shutil
17
 
 
 
18
  # Images
19
  torch.hub.download_url_to_file('https://cdn.pixabay.com/photo/2021/08/04/14/16/tower-6521842_1280.jpg', 'tower.jpg')
20
  torch.hub.download_url_to_file('https://cdn.pixabay.com/photo/2017/08/31/05/36/buildings-2699520_1280.jpg', 'city.jpg')
 
15
  from datetime import datetime, timedelta
16
  import shutil
17
 
18
+ print(torch.cuda.is_available())
19
+
20
  # Images
21
  torch.hub.download_url_to_file('https://cdn.pixabay.com/photo/2021/08/04/14/16/tower-6521842_1280.jpg', 'tower.jpg')
22
  torch.hub.download_url_to_file('https://cdn.pixabay.com/photo/2017/08/31/05/36/buildings-2699520_1280.jpg', 'city.jpg')