update

app.py

@@ -39,7 +39,10 @@ def make_clickable_model(model_name, link=None):
     #adding functionality for demo
     user_model_nm = model_name.split('/')[-1]
     print(f"user and model name is - {user_model_nm}")
-    model_nm = " ".join(user_model_nm.split('-'))
+    if '-' in user_model_nm:
+        model_nm = " ".join(user_model_nm.split('-'))
+    else:
+        model_nm = user_model_nm
     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')