chore: add descriptions

app.py

@@ -29,28 +29,42 @@ def predict(image: Image.Image) -> list[dict[str, float]]:
 
 def construct_ui():
     with gr.Blocks() as ui:
-        with gr.Row():
-            with gr.Column():
-                input_image = gr.Image(
-                    label="Input Image",
-                    type="pil",
-                    source="upload",
-                    interactive=True,
-                )
-
-                submit_btn = gr.Button(
-                    value="Submit",
-                    variant="primary",
-                )
-
-            with gr.Column():
-                result_salt = gr.Label(
-                    label=MODEL_SALT_BESTIMAGE,
-                )
-
-                result_cafe = gr.Label(
-                    label=MODEL_CAFE_AESTHETIC,
-                )
+        gr.Markdown(
+            """
+
+        # Waifu Aesthetics
+
+        Original model repos:
+
+        - https://huggingface.co/saltacc/beit-bestimage-salt
+        - https://huggingface.co/cafeai/cafe_aesthetic
+        
+        """
+        )
+
+        with gr.Column():
+            with gr.Row():
+                with gr.Column():
+                    input_image = gr.Image(
+                        label="Input Image",
+                        type="pil",
+                        source="upload",
+                        interactive=True,
+                    )
+
+                    submit_btn = gr.Button(
+                        value="Submit",
+                        variant="primary",
+                    )
+
+                with gr.Column():
+                    result_salt = gr.Label(
+                        label=MODEL_SALT_BESTIMAGE,
+                    )
+
+                    result_cafe = gr.Label(
+                        label=MODEL_CAFE_AESTHETIC,
+                    )
 
         submit_btn.click(
             predict, inputs=[input_image], outputs=[result_salt, result_cafe]