serdaryildiz commited on
Commit
25c03dd
1 Parent(s): 094d615

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,8 +8,8 @@ from Model import TRCaptionNet, clip_transform
8
 
9
  model_ckpt = "./checkpoints/TRCaptionNet_L14_berturk_tasviret.pth"
10
 
11
- device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
12
- # device = "cpu"
13
 
14
  preprocess = clip_transform(224)
15
  model = TRCaptionNet({
 
8
 
9
  model_ckpt = "./checkpoints/TRCaptionNet_L14_berturk_tasviret.pth"
10
 
11
+ # device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
12
+ device = "cpu"
13
 
14
  preprocess = clip_transform(224)
15
  model = TRCaptionNet({