Update README.md

README.md

@@ -67,16 +67,14 @@ For simplicity, you can run this here:
 Alternatively, embed the following code in your application:
 
 ```python
-import torch
-from transformers import AutoTokenizer, AutoConfig
-
-# Load the tokenizer and model from the local directory
-model_dir = "gsar78/HellenicSentimentAI_v2"
-tokenizer = AutoTokenizer.from_pretrained(model_dir)
-config = AutoConfig.from_pretrained(model_dir)
-model = CustomModel.from_pretrained(model_dir, config=config, num_emotion_labels=18)
+from transformers import AutoConfig, AutoModelForSequenceClassification, AutoTokenizer
 
+model_name = "gsar78/HellenicSentimentAI_v2"
 
+# Load the configuration, tokenizer, and model
+config = AutoConfig.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForSequenceClassification.from_pretrained(model_name)
 
 # Function to predict sentiment and emotion
 def predict(texts):