update new layout

app.py

@@ -8,13 +8,41 @@ import gradio as gr
 import pandas as pd
 from huggingface_hub import list_models
 
-# %% app.ipynb 1
+from diffusers import StableDiffusionPipeline
+
+submissions_list = list_models(filter=["dreambooth-hackathon", category], full=True)
+spaces_pipeline_load = [submission.id for submission in submissions_list ]
+for ids in spaces_pipeline_load:
+    mydict[ids] = StableDiffusionPipeline.from_pretrained(ids) #('ashiqabdulkhader/shiba-dog')
+    #f"pipeline{ids.split('//')[-1]}" = StableDiffusionPipeline.from_pretrained(ids) #('ashiqabdulkhader/shiba-dog')
+
+#pipeline = StableDiffusionPipeline.from_pretrained("ashiqabdulkhader/shiba-dog") #('pharma/sugar-glider')
+#image = pipeline().images[0]
+#image
+#https://huggingface.co/ashiqabdulkhader/shiba-dog
+    def filter_species(species):
+        return gr.Dropdown.update(choices=species_map[species], value=species_map[species][1]), gr.update(visible=True)
+
+    # %% app.ipynb 1
+def make_clickable_demo(model_name, prompt): #link=None):
+    #if link is None:
+    #    link = "https://huggingface.co/" + model_name
+    # Remove user from model name
+    prompt = "a photo of " + ' '.join(model_name.split('/')[-1].split['-']) + str(prompt)
+        
+    return gr.Button.update()
+    
 def make_clickable_model(model_name, link=None):
     if link is None:
         link = "https://huggingface.co/" + model_name
-    # Remove user from model name
-    return f'<a target="_blank" href="{link}">{model_name.split("/")[-1]}</a>'
 
+    #adding functionality for demo
+    prompt = "a photo of " + ' '.join(model_name.split('/')[-1].split['-']) + str(prompt)
+    pipeline = StableDiffusionPipeline.from_pretrained(model_name) #("ashiqabdulkhader/shiba-dog") #('pharma/sugar-glider')
+    image_demo = pipeline().images[0]
+    
+    # Remove user from model name
+    return image_out.Update(value=image_demo, label=model_name.split("/")[-1])  #f'<a target="_blank" href="{link}">{model_name.split("/")[-1]}</a>'
 
 def make_clickable_user(user_id):
     link = "https://huggingface.co/" + user_id
@@ -55,6 +83,9 @@ with block:
     For details on how to participate, check out the hackathon's guide [here](https://github.com/huggingface/diffusion-models-class/blob/main/hackathon/README.md).
     """
     )
+    with gr.Row():
+        prompt_in = gr.Textbox(label="Type in a Prompt. This will be suffixed to 'a photo of <model name>', so prompt accordingly -")
+        #button_in = gr.Button(label = "Generate Image using this model")
     with gr.Tabs():
         with gr.TabItem("Animal 🐨"):
             with gr.Row():
@@ -66,6 +97,8 @@ with block:
                 data_run.click(
                     get_submissions, inputs=gr.Variable("animal"), outputs=animal_data
                 )
+
+                
         with gr.TabItem("Science 🔬"):
             with gr.Row():
                 science_data = gr.components.Dataframe(
@@ -110,6 +143,11 @@ with block:
                     inputs=gr.Variable("wildcard"),
                     outputs=wildcard_data,
                 )
+    
+    with gr.Row() as your_model_demo :
+        image_out = gr.Image()
+        
+    button_in.click(make_clickable_demo, prompt_in, your_model_demo)
 
     block.load(get_submissions, inputs=gr.Variable("animal"), outputs=animal_data)
     block.load(get_submissions, inputs=gr.Variable("science"), outputs=science_data)