Spaces:
Runtime error
Runtime error
Commit
·
f8f1995
1
Parent(s):
26b285e
Change model model to eval
Browse files
app.py
CHANGED
@@ -22,6 +22,7 @@ for foundation in FOUNDATIONS:
|
|
22 |
pretrained_model_name_or_path=f"joshnguyen/mformer-{foundation}",
|
23 |
use_auth_token=True
|
24 |
)
|
|
|
25 |
MODELS[foundation] = model.to(DEVICE)
|
26 |
|
27 |
|
|
|
22 |
pretrained_model_name_or_path=f"joshnguyen/mformer-{foundation}",
|
23 |
use_auth_token=True
|
24 |
)
|
25 |
+
model.eval()
|
26 |
MODELS[foundation] = model.to(DEVICE)
|
27 |
|
28 |
|