updt

app.py

@@ -12,16 +12,16 @@ 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')
+#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)
+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):
@@ -39,7 +39,7 @@ def make_clickable_model(model_name, link=None):
     #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]
+    image_demo = pipeline(prompt).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>'