Update app.py

app.py

@@ -50,7 +50,7 @@ def generate(prompt, option, progress=gr.Progress()):
     def inference_callback(p, i, t, kwargs):
         progress((i+1, step))
         return kwargs
-    results = pipe(prompt, num_inference_steps=step, guidance_scale=0, callback_on_step_end=inference_callback, output_type="pil")
+    results = pipe(prompt, num_inference_steps=step, guidance_scale=0, callback_on_step_end=inference_callback, output_type="pt")
 
     # Safety check
     feature_extractor_input = image_processor.postprocess(results.images, output_type="pil")