Karim-Gamal
/

Roberta_finetuned_fake_news_english.pt

Text Classification

Inference Endpoints

Model card Files Files and versions Community

Karim-Gamal commited on Mar 16, 2023

Commit

4a78464

•

1 Parent(s): 77aefab

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -40,7 +40,7 @@ model = AutoModelForSequenceClassification.from_pretrained(model_name)
 # Tokenize the input text
 input_text = "This is a fake news article"
-inputs = tokenizer.encode_plus(input_text, padding=True, truncation=True, max_length=128, return_tensors="pt")
 # Get the model's prediction
 outputs = model(inputs["input_ids"], attention_mask=inputs["attention_mask"])

 # Tokenize the input text
 input_text = "This is a fake news article"
+inputs = tokenizer.encode_plus(input_text, padding=True, truncation=True, max_length=512, return_tensors="pt")
 # Get the model's prediction
 outputs = model(inputs["input_ids"], attention_mask=inputs["attention_mask"])