tweaks

app.py

@@ -78,7 +78,7 @@ def getModel(model):
 	if 'diffusion' in model_id:
 		pipe = DiffusionPipeline.from_pretrained(model_id)
 		pipe = pipe.to("cpu")
-		image = pipe(generator= torch.manual_seed(42), num_inference_steps=40).images[0]
+		image = pipe(generator= torch.manual_seed(42), num_inference_steps=1).images[0]
 	else:
 		pipe = DiffusionPipeline.from_pretrained('nathanReitinger/MNIST-diffusion')
 		pipe = pipe.to("cpu")
@@ -186,9 +186,9 @@ def getModel(model):
 
 
 	############################################ return image that you just generated
-	return (image, lowest_image)
+	return [image, lowest_image]
 
 
 import gradio as gr
-interface = gr.Interface(fn=getModel, inputs=[gr.Dropdown(modellist)], css="#output_image{width: 256px !important; height: 256px !important;}", outputs="image",  title='Model Problems (infringement)')
+interface = gr.Interface(fn=getModel, inputs=[gr.Dropdown(modellist)], css="#output_image{width: 256px !important; height: 256px !important;}", outputs=gr.outputs.Carousel("image"),  title='Model Problems (infringement)') # outputs="image",
 interface.launch()