Spaces:
Runtime error
Runtime error
Set eval mod
Browse files
app.py
CHANGED
@@ -24,6 +24,7 @@ def load():
|
|
24 |
processor = FlavaProcessor
|
25 |
prefix = "facebook/"
|
26 |
models[name] = model.from_pretrained(f"{prefix}{name}")
|
|
|
27 |
processors[name] = processor.from_pretrained(f"{prefix}{name}")
|
28 |
embeddings[name] = {
|
29 |
0: np.load(f"embeddings-{name}.npy"),
|
|
|
24 |
processor = FlavaProcessor
|
25 |
prefix = "facebook/"
|
26 |
models[name] = model.from_pretrained(f"{prefix}{name}")
|
27 |
+
models[name].eval()
|
28 |
processors[name] = processor.from_pretrained(f"{prefix}{name}")
|
29 |
embeddings[name] = {
|
30 |
0: np.load(f"embeddings-{name}.npy"),
|