aps commited on
Commit
aa4fb95
·
1 Parent(s): 73d70e7

Set eval mod

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