serdaryildiz commited on
Commit
e5829b0
1 Parent(s): 5f49231

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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
- 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"
 
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"