updt

app.py

@@ -37,7 +37,11 @@ def make_clickable_model(model_name, link=None):
         link = "https://huggingface.co/" + model_name
 
     #adding functionality for demo
-    prompt = "a photo of " + ' '.join(model_name.split('/')[-1].split['-']) + str(prompt)
+    user_model_nm = model_name.split('/')[-1]
+    print(f"user and model name is - {user_model_nm}")
+    model_nm = " ".join(user_model_nm.split('-'))
+    print(f"model name is - {model_nm}")
+    prompt = "a photo of " + model_nm + str(prompt)
     pipeline = StableDiffusionPipeline.from_pretrained(model_name) #("ashiqabdulkhader/shiba-dog") #('pharma/sugar-glider')
     image_demo = pipeline(prompt).images[0]