Update Gradio app

app.py

@@ -1,4 +1,3 @@
-import argparse
 import random
 import torch
 from rapgpt.config import Config
@@ -44,14 +43,29 @@ if __name__ == "__main__":
     gradio_app = gr.Interface(
         predict,
         inputs=[
-            gr.Textbox(value="ekip"),
-            gr.Number(value=100),
+            gr.Textbox(
+                value="ekip",
+                label="Lyrics prompt",
+                info="rapGPT will continue this prompt",
+            ),
+            gr.Number(
+                value=100,
+                maximum=100,
+                label="New tokens to generate",
+                info="Number of new tokens to generate (limited to 100)",
+            ),
             gr.Dropdown(
-                value="freeze corleone", choices=artists_tokens.keys(), type="index"
+                value="freeze corleone",
+                choices=artists_tokens.keys(),
+                type="index",
+                label="Artist",
+                info="Which artist style to generate",
+            ),
+            gr.Number(
+                value=42, label="Random seed", info="Change for different results"
             ),
-            gr.Number(value=42),
         ],
-        outputs=[gr.TextArea()],
+        outputs=[gr.TextArea(label="Generated Lyrics")],
         title="rapGPT",
     )