Spaces:

joshnguyen
/

mformer

Runtime error

joshnguyen commited on Sep 21, 2023

Commit

f8f1995

1 Parent(s): 26b285e

Change model model to eval

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,6 +22,7 @@ for foundation in FOUNDATIONS:
         pretrained_model_name_or_path=f"joshnguyen/mformer-{foundation}",
         use_auth_token=True
     )
     MODELS[foundation] = model.to(DEVICE)

         pretrained_model_name_or_path=f"joshnguyen/mformer-{foundation}",
         use_auth_token=True
     )
+    model.eval()
     MODELS[foundation] = model.to(DEVICE)