Update app.py

app.py

@@ -63,6 +63,8 @@ model = load_model(R"GPT-Shakespeare.pth")
 # Define the Gradio interface
 demo = gr.Interface(
     fn=generate_text,
+    title="Text Generation using GPT",
+    description = "<p style='text-align: center'> Decoder only transformer trained on \"Coriolanus\" by William Shakespeare </p>",
     inputs= [
         gr.Textbox(
             label="Input Text", 
@@ -83,18 +85,15 @@ demo = gr.Interface(
     outputs=gr.Textbox(lines=5, 
                        placeholder="Generated text will be shown here", 
                        label="Generated Text"),
-    title= """<h1 style='text-align: center;'>  Text Generation using GPT   \
-        <a href='https://github.com/KD1994/session-12-Transformer-from-scratch-pt2' target='_blank'> \
+    article= """<a href='https://github.com/KD1994/session-12-Transformer-from-scratch-pt2' target='_blank'> \
             <i class='fab fa-github' style='font-size: 24px;'></i></a> \
-            </h1> \
-                <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.0.0-beta3/css/all.min.css">""",
-    description = "<p style='text-align: center'> Decoder only transformer trained on \"Coriolanus\" by William Shakespeare </p>",
-    # examples = [
-    #     ["My noble Coriolanus, temper thy rage. These men hold"], 
-    #     ["Wisdom, say’st thou? Counsel, and truth? Nay, Menenius, they are"],
-    #     ["What speaks this man of war and violence?"],
-    #     ["Enough, Coriolanus! Thy words grow wild. What wouldst thou have?"]
-    # ]
+            <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.0.0-beta3/css/all.min.css">""",
+    examples = [
+        ["My noble Coriolanus, temper thy rage. These men hold"], 
+        ["Wisdom, say’st thou? Counsel, and truth? Nay, Menenius, they are"],
+        ["What speaks this man of war and violence?"],
+        ["Enough, Coriolanus! Thy words grow wild. What wouldst thou have?"]
+    ]
 )
 
 # Add error handling to launch