Spaces:
Runtime error
Runtime error
Ahsen Khaliq
commited on
Commit
•
dfa4a23
1
Parent(s):
77d98c4
Update app.py
Browse files
app.py
CHANGED
@@ -21,7 +21,8 @@ print(torch.cuda.is_available())
|
|
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')
|
23 |
idx = 0
|
24 |
-
|
|
|
25 |
def to_PIL_img(img):
|
26 |
result = Image.fromarray((img.data.cpu().numpy().transpose((1, 2, 0)) * 255).astype(np.uint8))
|
27 |
return result
|
|
|
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')
|
23 |
idx = 0
|
24 |
+
|
25 |
+
os.system("gdown https://drive.google.com/uc?id=1NDD54BLligyr8tzo8QGI5eihZisXK1nq")
|
26 |
def to_PIL_img(img):
|
27 |
result = Image.fromarray((img.data.cpu().numpy().transpose((1, 2, 0)) * 255).astype(np.uint8))
|
28 |
return result
|