Create main.py

main.py

@@ -0,0 +1,10 @@
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+
+tokenizer = AutoTokenizer.from_pretrained("vikram71198/distilroberta-base-finetuned-fake-news-detection")
+model = AutoModelForSequenceClassification.from_pretrained("vikram71198/distilroberta-base-finetuned-fake-news-detection")
+text = "This movie is really good!"
+inputs = tokenizer(text, return_tensors="pt")
+outputs = model(**inputs)
+
+print(f"Input text: {text}")
+print(f"Predicted label: {outputs[0]['label']}, score: {outputs[0]['score']:.2f}")