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")
|