Spaces:

johnpaulbin
/

beanbox-toxic-demo

Runtime error

johnpaulbin commited on Dec 16, 2023

Commit

955a6c8

•

1 Parent(s): 352195e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -98,7 +98,7 @@ model = nn.Sequential(
     nn.Dropout(0.05)  # Dropout
 )
-model.load_state_dict(torch.load("large.pth"), map_location=torch.device('cpu'))
 model.eval()

     nn.Dropout(0.05)  # Dropout
 )
+model.load_state_dict(torch.load("large.pth", map_location=torch.device('cpu')))
 model.eval()