Spaces:

Vageesh1
/

clip_gpt2

Sleeping

Vageesh1 commited on Sep 12, 2023

Commit

99b28b8

•

1 Parent(s): d21ae3c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -44,7 +44,7 @@ model = model.eval()
 coco_model = ClipCaptionModel(prefix_length)
 coco_model.load_state_dict(torch.load('COCO_model.h5',map_location=torch.device('cpu')),strict=False)
-# model = model.eval()
 def ui():

 coco_model = ClipCaptionModel(prefix_length)
 coco_model.load_state_dict(torch.load('COCO_model.h5',map_location=torch.device('cpu')),strict=False)
+model = model.eval()
 def ui():