Spaces:
Runtime error
Runtime error
Set eval mode
Browse files
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")
|