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