Update app.py

app.py

@@ -1,10 +1,38 @@
 import gradio as gr
-from task import tasks_config
-from pipeline_utils import handle_task_change, review_training_choices, test_pipeline
-from playground_utils import create_playground_header, create_playground_footer, create_tabs_header
+from pipeline_utils import task_dropdown_choices, handle_task_change, review_training_choices, test_pipeline
 
 playground = gr.Blocks()
 
+
+def create_playground_header():
+    gr.Markdown("""
+                # 🤗 Hugging Face Playground
+                **Try your ideas here. Select from Text, Image or Audio**
+                """)
+
+
+def create_playground_footer():
+    gr.Markdown("""
+                ### To Learn More about 🤗 Hugging Face,[Click Here](https://huggingface.co/docs)
+                ### [Click Here](https://huggingface.co/spaces/nsethi610/ns-gradio-apps/discussions/1) to provide Feedback, or participate in development of this tool.Let's make AI easy for everyone.
+                """)
+
+
+def create_tabs_header():
+    with gr.Row():
+        with gr.Column(scale=4):
+            radio = gr.Radio(
+                ["Use Pipeline", "Fine Tune"],
+                label="Select Use Pipeline to try out HF models or Fine Tune to test it on your own datasets",
+                value="Use Pipeline",
+                interactive=True,
+            )
+        with gr.Column(scale=1):
+            test_pipeline_button = gr.Button(
+                value="Test", variant="primary", size="sm")
+        return radio, test_pipeline_button
+
+
 with playground:
     create_playground_header()
     with gr.Tabs():
@@ -13,8 +41,7 @@ with playground:
             with gr.Row(visible=True) as use_pipeline:
                 with gr.Column():
                     task_dropdown = gr.Dropdown(
-                        choices=[(task["name"], task_id)
-                                 for task_id, task in tasks_config.items()],
+                        choices=task_dropdown_choices(),
                         label="Task",
                         interactive=True,
                         info="Select Pipelines for natural language processing tasks or     type if you have your own."
@@ -44,7 +71,8 @@ with playground:
                          inputs=radio, outputs=use_pipeline)
             test_pipeline_button.click(test_pipeline,
                                        inputs=[
-                                           task_dropdown, model_dropdown, prompt_textarea, context_for_question_answer],
+                                           task_dropdown, model_dropdown, prompt_textarea,
+                                           context_for_question_answer],
                                        outputs=text)
         with gr.TabItem("Image"):
             radio, test_pipeline_button = create_tabs_header()