Update app.py

app.py

@@ -37,6 +37,8 @@ pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
     safety_checker=None,
 )
 
+pipe.enable_model_cpu_offload()
+pipe.enable_xformers_memory_efficient_attention()
 
 # inference function takes prompt, negative prompt and image
 def infer(prompts, negative_prompts, image):
@@ -80,7 +82,7 @@ description = "Sketch your game battlemap and add some prompts to let the magic
 gr.Interface(
     fn=infer,
     inputs=["text", "text", "image"],
-    outputs="gallery",
+    outputs="image",
     title=title,
     description=description,
 ).launch()