aps commited on
Commit
7eed789
1 Parent(s): 4a76f18

Set eval mode

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -34,6 +34,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
34
  model = FlavaModel.from_pretrained("facebook/flava-full")
35
  tokenizer = BertTokenizer.from_pretrained("facebook/flava-full")
36
  model = model.to(device)
 
37
 
38
  # Load Data
39
  photo_features = np.load("unsplash-dataset/features.npy")
 
34
  model = FlavaModel.from_pretrained("facebook/flava-full")
35
  tokenizer = BertTokenizer.from_pretrained("facebook/flava-full")
36
  model = model.to(device)
37
+ model.eval()
38
 
39
  # Load Data
40
  photo_features = np.load("unsplash-dataset/features.npy")