Update README.md

README.md

@@ -77,8 +77,28 @@ pipe = pipeline("text-classification", model="Zabihin/Symptom_to_Diagnosis")
 Example:
 result = pipe("I've been having headaches and migraines, and I can't sleep. My whole body shakes and twitches. Sometimes I feel lightheaded.")
 result:
+
 [{'label': 'drug reaction', 'score': 0.9489321112632751}]
 
+or 
+
+
+from transformers import pipeline
+
+# Load the model
+classifier = pipeline("text-classification", model="Zabihin/Symptom_to_Diagnosis", tokenizer="Zabihin/Symptom_to_Diagnosis")
+
+# Example input text
+input_text = "I've been having headaches and migraines, and I can't sleep. My whole body shakes and twitches. Sometimes I feel lightheaded."
+
+# Get the predicted label
+result = classifier(input_text)
+
+# Print the predicted label
+predicted_label = result[0]['label']
+print("Predicted Label:", predicted_label)
+
+Predicted Label: drug reaction