joshnguyen commited on
Commit
f8f1995
·
1 Parent(s): 26b285e

Change model model to eval

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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