Update app.py

app.py

@@ -10,8 +10,20 @@ repo = git.Repo.clone_from(url='https://huggingface.co/Liyulingyue/Neolle_Face_G
 model_path = "dream_outputs"
 pipe = StableDiffusionPipeline.from_pretrained(model_path)
 
-def generate_images(prompt="Neolle", num_inference_steps=50, guidance_scale=7.5):
-    image = pipe(prompt, num_inference_steps=int(num_inference_steps),guidance_scale=float(guidance_scale)).images[0]
+def generate_images(prompt="Neolle", num_inference_steps, guidance_scale):
+    # num_inference_steps to number
+    try:
+        infer_steps = int(num_inference_steps)
+    except:
+        infer_steps = 50
+
+    # guidance_scale to number
+    try:
+        gui_scale = float(guidance_scale)
+    except:
+        gui_scale = 7.5
+    
+    image = pipe(prompt, num_inference_steps=infer_steps,guidance_scale=gui_scale).images[0]
     # image = os.getcwd()
     return image