Spaces:

johnpaulbin
/

beanbox-toxicity

Runtime error

johnpaulbin commited on Dec 16, 2023

Commit

aea79b5

•

1 Parent(s): 1877179

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -100,7 +100,7 @@ model = nn.Sequential(
     nn.Dropout(0.05)  # Dropout
 )
-model.load_state_dict(torch.load("large.pth"), map_location=torch.device('cpu'))
 model.eval()
 @app.route('/infer', methods=['POST'])

     nn.Dropout(0.05)  # Dropout
 )
+model.load_state_dict(torch.load("large.pth", map_location=torch.device('cpu')))
 model.eval()
 @app.route('/infer', methods=['POST'])