Update app.py

app.py

@@ -26,6 +26,38 @@ LINKS = {
     "if-v1-0": "https://huggingface.co/DeepFloyd/IF-I-XL-v1.0",
     "karlo": "https://huggingface.co/kakaobrain/karlo-v1-alpha",
 }
+SD_1_5_RESULT = """
+![img](https://huggingface.co/datasets/OpenGenAI/logos/blob/main/37351293.png)
+"The traditional one 🔥! You mostly resonate with **Stable Diffusion 1-5** released by RunwayML.
+\n Stable Diffusion 1-5 is the most used open-source text-to-image model offering an amazing speed-to-image-quality trade-off!
+Check out your soulmate [here](https://huggingface.co/runwayml/stable-diffusion-v1-5).
+"""
+SD_2_1_RESULT = """
+![img](https://huggingface.co/datasets/OpenGenAI/logos/resolve/main/7vmYr2XwVcPtkLzac_jxQ.png)
+"The creative one 🎨! You mostly resonate with **Stable Diffusion 2-1** released by Stability AI.
+\n Stable Diffusion 2-1 is the latest open-source release of Stable Diffusion and allows to render stunning images of much larger sizes than Stable Diffusion v1.
+Try it out [here](https://huggingface.co/stabilityai/stable-diffusion-2-1).
+"""
+IF_V1_0 = """
+![img](https://huggingface.co/datasets/OpenGenAI/logos/blob/main/1662453741854-63170739dc97a974718be2c7.png)
+The powerful one ⚡! You mostly resonate with **IF v1-0** released by DeepFloyd.
+\n IF v1-0 is by far the largest of the open-sourced text-to-image models and is a very powerful image generator.
+Besides being able to generate multiple complex concepts in the same image, IF v1-0 is also extremely good at generating text in images.
+Check out your new best friend [here](https://huggingface.co/DeepFloyd/IF-I-XL-v1.0)
+"""
+KARLO = """
+![img](https://huggingface.co/datasets/OpenGenAI/logos/blob/main/1670220967262-615ed619c807b26d117a49bd.png)
+The precise one 🎯! You mostly resonate with **Karlo** released by KakaoBrain.
+\n Karlo is based on the same architecture as DALLE-2 and has been trained on the [well curated COYO dataset](https://huggingface.co/datasets/kakaobrain/coyo-700m).
+Play around with it [here]("https://huggingface.co/kakaobrain/karlo-v1-alpha").
+"""
+
+RESULT = {
+    "sd-v1-5": SD_1_5_RESULT,
+    "sd-v2-1": SD_2_1_RESULT,
+    "if-v1-0": IF_V1_0,
+    "karlo": KARLO,
+}
 NUM_QUESTIONS = 10
 MODEL_KEYS = "-".join(SUBMISSIONS.keys())
 SUBMISSION_ORG = f"results-{MODEL_KEYS}"
@@ -170,7 +202,7 @@ def change_view(row_number, dataframe):
             intro_view: gr.update(visible=True),
             result_view: gr.update(visible=True),
             gallery_view: gr.update(visible=False),
-            result: f"You are of type: [**{favorite_model}**]({LINKS[favorite_model]}) 🔥",
+            result: RESULT[favorite_model],
         }
     else:
         return {
@@ -181,19 +213,27 @@ def change_view(row_number, dataframe):
         }
 
 
-TITLE = "# Open Parti Prompts - Who is your open-source genAI model?"
+TITLE = "# What's Your AI Gen Personality 🧬?"
+
 DESCRIPTION = """
-*This is an interactive game in which you click through pre-generated images from SD-v1-5, SD-v2.1, Karlo, and IF 
-using [Parti Prompts](https://huggingface.co/datasets/nateraw/parti-prompts) prompts.* \n
-*You choices will go into the official [Parti Prompts Leaderboard](https://huggingface.co/spaces/OpenGenAI/parti-prompts-leaderboard) ❤️.*
+*In this interactive game you are shown image descriptions along side 4 AI generated images. Select the image that best fits the image description.
+Answer **10** questions to find out what AI generator most resonates with you. If none of the images fit the prompt, choose the image that comes closest 
+in your opinion.* \n\n
 """
 EXPLANATION = """\n\n
-## How it works 📖 \n\n
+## What to do 📖 \n\n
 
 1. Click on 'Start'
 2. A prompt and 4 different images are displayed
 3. Select your favorite image
-4. After 10 rounds your favorite diffusion model is displayed
+4. After 10 rounds your favorite diffusion model is displayed \n\n
+
+*The prompts you are shown originate from the [Parti Prompts](https://huggingface.co/datasets/nateraw/parti-prompts) dataset. 
+Parti Prompts is designed to test text-to-image AI models on 1600+ prompts of varying difficulty and categories. 
+The images you are shown have been pre-generated with 4 state-of-the-art open-sourced text-to-image models. \n\n
+You answers will be used to contribute to the official [**Open Parti Prompts Leaderboard**](https://huggingface.co/spaces/OpenGenAI/parti-prompts-leaderboard). \n
+By playing this game, you are greatly helping the community to have a better, **human-preference-aligned** metric to compare text-to-image models ❤️. \n\n
+Every month, the generated images will be updated with possibly improved models.*
 """
 
 GALLERY_COLUMN_NUM = len(SUBMISSIONS)