update docs

app.py

@@ -223,8 +223,6 @@ if __name__ == "__main__":
         gr.Markdown(
             """
 # Generative Augmented Image Classifiers
-This demo showcases the performance of image classifiers trained on various datasets as part of the project `Investigating the Effectiveness of Generative Diffusion Models in Synthesizing Images for Data Augmentation in Image Classification' dissertation.
-
 Main GitHub Repo: [Generative Data Augmentation](https://github.com/zhulinchng/generative-data-augmentation) | Generative Data Augmentation Demo: [Generative Data Augmented](https://huggingface.co/spaces/czl/generative-data-augmentation-demo).
 """
         )
@@ -267,9 +265,10 @@ Main GitHub Repo: [Generative Data Augmentation](https://github.com/zhulinchng/g
                     inputs=[methods, dataset_type],
                     outputs=[training_ds],
                 )
-                generate_button = gr.Button("Sample Random Image")
                 random_image_output = gr.Image(type="pil", label="Image to Classify")
-                classify_button_random = gr.Button("Classify")
+                with gr.Row():
+                    generate_button = gr.Button("Sample Random Image")
+                    classify_button_random = gr.Button("Classify")
             with gr.Column():
                 output_label_random = gr.Label(num_top_classes=5)
                 download_model = gr.DownloadButton(
@@ -298,6 +297,15 @@ Main GitHub Repo: [Generative Data Augmentation](https://github.com/zhulinchng/g
                     inputs=[dataset_type, arch_type, methods, training_ds],
                     outputs=[download_model],
                 )
+                gr.Markdown(
+                    """
+This demo showcases the performance of image classifiers trained on various datasets as part of the project 'Investigating the Effectiveness of Generative Diffusion Models in Synthesizing Images for Data Augmentation in Image Classification' dissertation.
+
+View the models and files used in this demo [here](https://huggingface.co/spaces/czl/generative-augmented-classifiers/tree/main).
+
+Usage Instructions & Documentation [here](https://huggingface.co/spaces/czl/generative-augmented-classifiers/blob/main/README.md).
+                """
+                )
 
         generate_button.click(
             get_random_image,