added progress bar for image generation

app.py

@@ -18,7 +18,7 @@ def get_model_list(category):
 
 
 def get_initial_prompt(model_nm):
-    #a photo of peterj/shbrcky-dog 
+    #Example - a photo of shbrcky dog 
     user_model_nm = model_nm.split('/')[-1]
     if '-' in user_model_nm:
         prompt = " ".join(user_model_nm.split('-'))
@@ -26,12 +26,19 @@ def get_initial_prompt(model_nm):
         prompt = user_model_nm
     return gr.Textbox.update(value="a photo of " + prompt + " ")
 
-def make_demo(model_name, prompt): #link=None):
-    #prompt = "a photo of " + ' '.join(model_name.split('/')[-1].split['-']) + str(prompt)
-    pipeline = StableDiffusionPipeline.from_pretrained(model_name) #("ashiqabdulkhader/shiba-dog") #('pharma/sugar-glider')
+def get_pipeline(model_name): #, progress=gr.Progress(track_tqdm=True)):
+    #Using diffusers pipeline to generate an image for the demo
+    #Loading Your Dreambooth model
+    pipeline = StableDiffusionPipeline.from_pretrained(model_name) # Example - ("ashiqabdulkhader/shiba-dog") or ('pharma/sugar-glider')
+    return pipeline 
+
+def make_demo(model_name, prompt, progress=gr.Progress(track_tqdm=True)):
+    #Using diffusers pipeline to generate an image for the demo
+    progress(0, desc="Starting...")
+    pipeline = get_pipeline(model_name) #StableDiffusionPipeline.from_pretrained(model_name) # Example - ("ashiqabdulkhader/shiba-dog") or ('pharma/sugar-glider')
+    #Generating Image from your prompt
     image_demo = pipeline(prompt).images[0]    
-    return image_demo #gr.Button.update()
-
+    return image_demo 
 
 def make_clickable_model(model_name, link=None):
     if link is None: