Update README.md

README.md

@@ -22,9 +22,12 @@ base_model:
 ```Python
 from transformers import pipeline
 
-
-emotion_classifier = pipeline("text-classification", model="XuehangCang/Emotion-Classification")
-
+# Ensure the model and tokenizer are loaded on the GPU by setting device=0
+emotion_classifier = pipeline(
+    "text-classification",
+    model="XuehangCang/Emotion-Classification",
+    # device=0  # Use the first GPU device
+)
 
 texts = [
     "I'm so happy today!",
@@ -35,7 +38,6 @@ texts = [
 
 for text in texts:
     result = emotion_classifier(text)
-    # 打印分类结果
     print(f"Text: {text}")
     print(f"Emotion classification result: {result}\n")