samyak152002
/

Tweet_Abortion_Analysis

Feature Extraction

text-embeddings-inference

Inference Endpoints

Model card Files Files and versions Community

samyak152002 commited on Jun 25, 2023

Commit

bc9b616

•

1 Parent(s): 02fd376

Update script

Files changed (1) hide show

script +2 -2

script CHANGED Viewed

@@ -2,8 +2,8 @@ import torch
 from transformers import DistilBertTokenizer, DistilBertModel
 # Load the tokenizer and model
-tokenizer = DistilBertTokenizer.from_pretrained("./directory")
-model = DistilBertModel.from_pretrained("./directory")
 # Define the inference function
 def predict(text):

 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):