Vageesh1 commited on
Commit
cbbcfd4
1 Parent(s): c44b109

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,12 +16,12 @@ prefix_length = 10
16
 
17
  model = ClipCaptionModel(prefix_length)
18
 
19
- model.load_state_dict(torch.load('C:\Deep learning lab\DLops Project\Cl+gpt2\model.h5',map_location=torch.device('cpu')))
20
 
21
  model = model.eval()
22
 
23
  coco_model = ClipCaptionModel(prefix_length)
24
- coco_model.load_state_dict(torch.load('C:\Deep learning lab\DLops Project\Cl+gpt2\COCO_model.h5',map_location=torch.device('cpu')))
25
  model = model.eval()
26
 
27
 
 
16
 
17
  model = ClipCaptionModel(prefix_length)
18
 
19
+ model.load_state_dict(torch.load('model.h5',map_location=torch.device('cpu')))
20
 
21
  model = model.eval()
22
 
23
  coco_model = ClipCaptionModel(prefix_length)
24
+ coco_model.load_state_dict(torch.load('COCO_model.h5',map_location=torch.device('cpu')))
25
  model = model.eval()
26
 
27