fix: fix the ambigious naming of "load_dataset"

app.py

@@ -29,7 +29,7 @@ def get_tweet(data):
   labels = [get_label(x['label']) for x in data]
   return tweets, labels
 
-def load_dataset():
+def get_train_val_test_sets():
     dataset = load_dataset("dair-ai/emotion")
     train = dataset['train']
     val = dataset['validation']
@@ -37,7 +37,7 @@ def load_dataset():
     return train, val, test
 
 
-train, val, test = load_dataset()
+train, val, test = get_train_val_test_sets()
 tweets, labels = get_tweet(train)
 tokenizer = Tokenizer(num_words=10000,oov_token='<UNK>')
 tokenizer.fit_on_texts(tweets)
@@ -83,7 +83,7 @@ def load_model():
     _ = model.evaluate(test_seq, test_labels)
     i = random.randint(0, len(test_labels) - 1)
     print('Sentence:', test_tweets[i])
-    print('Emotion:', index_to_class[test_labels[i]])
+    print('Emotion:', index_to_class[test_labels[i]]) 
     p = model.predict(np.expand_dims(test_seq[i],axis=0))[0]
     pred_class = index_to_class[np.argmax(p).astype('uint8')]