Update app.py

app.py

@@ -41,8 +41,9 @@ def generate_image(prompt, ckpt):
         loaded = num_inference_steps
 
     if num_inference_steps == 1:
-        timesteps = [399]
-    results = pipe(prompt, num_inference_steps=num_inference_steps, guidance_scale=0, timesteps=timesteps) 
+        results = pipe(prompt, num_inference_steps=num_inference_steps, guidance_scale=0, timesteps=[399])
+    else:
+        results = pipe(prompt, num_inference_steps=num_inference_steps, guidance_scale=0)
 
 
     return results.images[0]