Update test_model.py

test_model.py

@@ -1,12 +1,16 @@
-from transformers import pipeline
+from transformers import pipeline, AutoTokenizer
 
-# Load the model from Hugging Face
-model_name = "DilipKY/my-text-classifier"  # Ensure this matches your model name
-classifier = pipeline("text-classification", model=model_name)
+model_name = "DilipKY/my-text-classifier"
+tokenizer_name = "bert-base-uncased"  # Use "bert-base-cased" if your model is case-sensitive
 
-# Test with a sample text
-sample_text = "This movie was amazing! The plot was so engaging and the acting was superb."
+# Load the correct tokenizer
+tokenizer = AutoTokenizer.from_pretrained(tokenizer_name)
+
+# Load the model with the tokenizer
+classifier = pipeline("text-classification", model=model_name, tokenizer=tokenizer)
+
+# Test the model
+sample_text = "I love this product! It's amazing."
 result = classifier(sample_text)
 
-# Print the output
 print(result)