samyak152002
/

Tweet_Abortion_Analysis

Feature Extraction

Inference Endpoints

Model card Files Files and versions Community

samyak152002 commited on Jun 25, 2023

Commit

4946849

·

1 Parent(s): 0cf557d

Update script

Files changed (1) hide show

script +1 -1

script CHANGED Viewed

@@ -3,7 +3,7 @@ from transformers import DistilBertTokenizer, DistilBertModel
 # Load the tokenizer and model
 tokenizer = DistilBertTokenizer.from_pretrained("tokenizer_config.json")
-model = DistilBertModel.from_pretrained("pytorch_model.bin")
 # Define the inference function
 def predict(text):

 # Load the tokenizer and model
 tokenizer = DistilBertTokenizer.from_pretrained("tokenizer_config.json")
+model = DistilBertModel.from_pretrained("model.pt")
 # Define the inference function
 def predict(text):