samyak152002
/

Tweet_Abortion_Analysis

Feature Extraction

Model card Files Files and versions

samyak152002 commited on Jun 25, 2023

Commit

5a3da59

·

1 Parent(s): a7fc3b1

Update script

Files changed (1) hide show

script +1 -1

script CHANGED Viewed

@@ -10,7 +10,7 @@ device = torch.device("cpu")
 model.to(device)
 # Load the saved model state dictionary
-model.load_state_dict(torch.load("path/to/save/directory/model.pt", map_location=device))
 # Set the model to evaluation mode
 model.eval()

 model.to(device)
 # Load the saved model state dictionary
+model.load_state_dict(torch.load("model.pt", map_location=device))
 # Set the model to evaluation mode
 model.eval()