Update app.py

app.py

@@ -41,7 +41,7 @@ def show_images_save(x):
 
 def generate(schedul):
     for i in model:
-        if schedul.name == i.name:
+        if schedul == i.name:
             scheduler = i.code
     scheduler.set_timesteps(num_inference_steps=40)
     x = torch.randn(1, 4, 64, 64).to(device)