Spaces:
Running
Running
serdaryildiz
commited on
Commit
•
e5829b0
1
Parent(s):
5f49231
Update app.py
Browse files
app.py
CHANGED
@@ -7,10 +7,10 @@ import torch
|
|
7 |
from Model import TRCaptionNet, clip_transform
|
8 |
|
9 |
model_ckpt = "./checkpoints/TRCaptionNet_L14_berturk.pth"
|
10 |
-
if not os.path.exists(model_ckpt):
|
11 |
-
|
12 |
-
|
13 |
-
|
14 |
|
15 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
16 |
# device = "cpu"
|
|
|
7 |
from Model import TRCaptionNet, clip_transform
|
8 |
|
9 |
model_ckpt = "./checkpoints/TRCaptionNet_L14_berturk.pth"
|
10 |
+
# if not os.path.exists(model_ckpt):
|
11 |
+
os.makedirs("./checkpoints/", exist_ok=True)
|
12 |
+
url = 'https://drive.google.com/u/0/uc?id=14Ll1PIQhsMSypHT34Rt9voz_zaAf4Xh9&export=download&confirm=t&uuid=9b4bf589-d438-4b4f-a37c-fc34b0a63a5d&at=AB6BwCAY8xK0EZiPGv2YT7isL8pG:1697575816291'
|
13 |
+
gdown.download(url, model_ckpt, quiet=False)
|
14 |
|
15 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
16 |
# device = "cpu"
|