Update main.py

main.py

@@ -19,8 +19,8 @@ def generate():
     seed = randrange(1, 9999999999)
     generator = torch.Generator('cuda').manual_seed(seed)
     image = pipe(text, num_inference_steps=25, generator=generator).images[0]
-    return Response(image, mimetype='image/png', direct_passthrough=True)
+    return send_file(image, mimetype='image/png')
 
 
 if __name__ == '__main__':
-    app.run('0.0.0.0', 8282, debug=True)
+    app.run('0.0.0.0', 8282, debug=False)