Spaces:

for876543
/

plant-id-3

Build error

for876543 commited on Oct 13, 2022

Commit

4edaa99

•

1 Parent(s): 705a628

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ class GELU(torch.nn.Module):
 torch.nn.modules.activation.GELU = GELU
 model = torch.load("/home/user/app/model_3nq5558t_scripted.pkl",map_location=torch.device('cpu'))
 with open('/home/user/app/val.json', 'r') as handle:
     parsed = json.load(handle)

 torch.nn.modules.activation.GELU = GELU
 model = torch.load("/home/user/app/model_3nq5558t_scripted.pkl",map_location=torch.device('cpu'))
+model.eval()
 with open('/home/user/app/val.json', 'r') as handle:
     parsed = json.load(handle)