Update README.md

README.md

@@ -18,7 +18,7 @@ tokenizer = AutoTokenizer.from_pretrained("Chirayu/mt5-multilingual-sentiment")
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = model.to(device)
 
-def get_subject(text, num_beams=2,max_length=512, repetition_penalty=2.5, length_penalty=1, early_stopping=True,top_p=.95, top_k=50, num_return_sequences=1):
+def get_sentiment(text, num_beams=2,max_length=512, repetition_penalty=2.5, length_penalty=1, early_stopping=True,top_p=.95, top_k=50, num_return_sequences=1):
   
   input_ids = tokenizer.encode(
     text, return_tensors="pt", add_special_tokens=True