Update app.py

app.py

@@ -167,7 +167,7 @@ def generate_with_all_models(commentary_text, num_words, max_length, temperature
     lstm_output = generate_with_lstm(commentary_text, num_words)
     rnn_output = generate_with_rnn(commentary_text, num_words)
     gpt2_finetuned_output, gpt2_base_output = generate_with_gpt2(commentary_text, max_length, temperature)
-    return  rnn_output, lstm_output, gpt2_finetuned_output, gpt2_base_output
+    return  rnn_output, lstm_output, gpt2_base_output, gpt2_finetuned_output
 
 # Create the Gradio interface
 iface = gr.Interface(
@@ -181,11 +181,11 @@ iface = gr.Interface(
     outputs=[
         gr.Textbox(label="RNN Model Output"),
         gr.Textbox(label="LSTM Model Output"),
-        gr.Textbox(label="GPT-2 Finetuned Model Output"), 
-        gr.Textbox(label="GPT-2 Base Model Output")
+        gr.Textbox(label="GPT-2 Base Model Output (not-finetuned)"), 
+        gr.Textbox(label="GPT-2 Finetuned Model Output")
     ],
     title="Next word prediction with RNN, LSTM and GPT-2 Models",
-    description="Start writing a cricket commentary and various models will try to continue it. Compare outputs from LSTM, RNN, and GPT-2 (finetuned and base) models."
+    description="Start writing a cricket commentary and various models will try to continue it. Compare outputs from LSTM, RNN, and GPT-2 (base & finetuned) models."
 )
 
 # Launch the app