Update app.py

app.py

@@ -42,11 +42,11 @@ def generate_image(prompt, ckpt="4-Step"):
         loaded = num_inference_steps
 
     if loaded == 1:
-        results = pipe(prompt, num_inference_steps=num_inference_steps, guidance_scale=0, timesteps=[399])
+        timesteps=[399]
     else:
-        results = pipe(prompt, num_inference_steps=num_inference_steps, guidance_scale=0, timesteps=[999, 749, 499, 249])
-
+        timesteps=[999, 749, 499, 249]
 
+    results = pipe(prompt, num_inference_steps=num_inference_steps, guidance_scale=0, timesteps=timesteps)
     return results.images[0]