Spaces:

LuyangZ
/

FloorAI

Runtime error

LuyangZ commited on May 24, 2024

Commit

bcf93ba

verified ·

1 Parent(s): ca64586

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -125,7 +125,7 @@ def floorplan_generation(outline, num_of_rooms):
         image = pipeline(validation_prompt,
                          validation_image,
-                         num_inference_steps=50,
                          generator=generator).images[0]
         image = np.array(image)

         image = pipeline(validation_prompt,
                          validation_image,
+                         num_inference_steps=20,
                          generator=generator).images[0]
         image = np.array(image)