The v1.2 commit

sgg_app.py

@@ -7,10 +7,6 @@ import random
 
 # Load the dataset
 dataset = load_dataset("SpartanCinder/song-lyrics-artist-classifier")
-# print(dataset.column_names)
-# print(dataset['train']['Artist'])
-# artist_list = list(set(dataset['train']['Artist']))
-# print(artist_list)
 
 def generate_song(state, language_model, generate_song):
 
@@ -54,8 +50,6 @@ def generate_song(state, language_model, generate_song):
         print(tokenizer.decode(encoded_output[0], skip_special_tokens=True))
         # But this output is repeating, so I need ot adjust this so that it is not repeating.
     elif language_model == "Custom Gpt2":
-        # tokenizer = AutoTokenizer.from_pretrained("SpartanCinder/GPT2-pretrained-lyric-generation")
-        # model = AutoModelForCausalLM.from_pretrained("SpartanCinder/GPT2-pretrained-lyric-generation")
         # encoded_output = model.generate(input_ids, max_length=max_length, num_beams=5, num_return_sequences=5, do_sample=False, no_repeat_ngram_size=2) # Generate text
         encoded_output = model.generate(input_ids, max_length=max_length, num_return_sequences=5, do_sample=True, top_p = 0.95, ) 
         # Decode output
@@ -71,9 +65,8 @@ def generate_song(state, language_model, generate_song):
     # Decode output
     output = tokenizer.decode(encoded_output[0], skip_special_tokens=True)
     # Remove the first line of the output if it contains newline characters
-    # if '\n' in output:
-    #     output = '\n'.join(output.split('\n')[1:])
-    # formatted_output = output.split('\n')[0] # might have to remove this line
+    if '\n' in output:
+        output = '\n'.join(output.split('\n')[1:])
     song_text = output
 
     # Generate the multiple-choice options
@@ -88,21 +81,6 @@ def generate_song(state, language_model, generate_song):
     return state, song_text, ', '.join(options)
 
 #Check the selected artist and return whether it's correct
-# def on_submit_answer(state, correct_choice, user_choice, submit_answer):
-#     if submit_answer:
-#         if not user_choice:
-#             return {"Error": "Please select an artist before submitting an answer."}
-#         # Check if 'correct_choice' is in the state keys
-#         if 'correct_choice' in state:
-#             correct_answer = state['correct_choice']
-#             if correct_answer == user_choice:
-#                 return {"Result": f"You guessed the right artist: {correct_choice}"}
-#             else:
-#                 return {"Result": f"You selected {user_choice}, but the correct answer is {correct_choice}"}
-#         else:
-#             print("The 'correct_choice' key does not exist in the state.")
-#             return None
-
 def on_submit_answer(state, user_choice):
     # Map the user's choice (A, B, C, or D) to an index
     choice_to_index = {'A': 0, 'B': 1, 'C': 2, 'D': 3}
@@ -125,10 +103,6 @@ def pick_artist(dataset):
     artist_choice = random.choice(artist_choice)
     return artist_choice
 
-        # print("The 'Artist' column does not exist in the dataset.")
-        # artist_choice = "Green Day"
-    # return artist_choice
-
 def generate_artist_options(dataset, correct_artist):
     # Generate 3 incorrect options
     all_artists = list(set(dataset['train']['Artist']))
@@ -141,22 +115,19 @@ def generate_artist_options(dataset, correct_artist):
 def generate_multiple_choice_check(options, correct_choice):
     return {option: option == correct_choice for option in options}
 
-def check_correct_choice(user_choice, correct_choice):
-    if user_choice == correct_choice:
-        return True
-    return user_choice == correct_choice
-
 with gr.Blocks(title="Song Generator Guessing Game") as game_interface:
+    gr.Markdown(" # Song Generator Guessing Game")
+    gr.Image("https://huggingface.co/spaces/SpartanCinder/NLP_Song_Generator_Guessing_Game/blob/main/RobotSinger.png")
     state = gr.State({'options': []})
     language_model = gr.Radio(["Custom Gpt2", "Gpt2-Medium", "facebook/bart-base","Gpt-Neo", "Customized Models"], label="Model Selection", info="Select the language model to generate the song.")
     generate_song_button = gr.Button("Generate Song")
     generated_song = gr.Textbox(label="Generated Song")
     artist_choice_display = gr.Textbox(interactive=False, label="Multiple-Choice Options")
     user_choice = gr.Radio(["A", "B", "C", "D"], label="Updated Options", info="Select the artist that you suspect is the correct artist for the song.")
-    # timer = gr.HTML("<div id='progress-bar' style='width: 100%; background-color: #f3f3f3; border: 1px solid #bbb;'><div id='progress' style='height: 20px; width: 0%; background-color: #007bff;'></div></div><script>function startTimer() {var time = 30; var timer = setInterval(function() {time--; document.getElementById('progress').style.width = (time / 30 * 100) + '%'; if (time <= 0) {clearInterval(timer);}}, 1000);}</script>", label="Timer")
     submit_answer_button = gr.Button("Submit Answer")
     correct_answer = gr.Textbox(label="Results")
     
+    
     generate_song_button.click(
         generate_song,
         [state, language_model, generate_song_button],