i

app.py

@@ -167,9 +167,10 @@ def greet(input):
 	#response2 = transform2(transform(np.moveaxis(samples.detach().cpu().numpy(),1,3)[1]))
 	#response = im.fromarray(intermediate[24][0][1]).convert("RGB")
 	response = transform2(transform(np.moveaxis(intermediate,2,4)[0][0]))
-	response2 = transform2(transform(np.moveaxis(intermediate,2,4)[12][0]))
-	response3 = transform2(transform(np.moveaxis(intermediate,2,4)[24][0]))
-	return response, response2, response3
+	response2 = transform2(transform(np.moveaxis(intermediate,2,4)[10][0]))
+	response3 = transform2(transform(np.moveaxis(intermediate,2,4)[20][0]))
+	response4 = transform2(transform(np.moveaxis(intermediate,2,4)[24][0]))
+	return response, response2, response3, response4
 
 transform2 = transforms.ToPILImage()
 
@@ -177,6 +178,7 @@ transform2 = transforms.ToPILImage()
 #iface.launch()
 
 #iface = gr.Interface(fn=greet, inputs=[gr.Textbox(label="Text to find entities", lines=2)], outputs=[gr.HighlightedText(label="Text with entities")], title="NER with dslim/bert-base-NER", description="Find entities using the `dslim/bert-base-NER` model under the hood!", allow_flagging="never", examples=["My name is Andrew and I live in California", "My name is Poli and work at HuggingFace"])
-iface = gr.Interface(fn=greet, inputs=[gr.Textbox(label="Co-Retailing Business")], outputs=[gr.outputs.Image(type="pil", width=64, label="Output Image"), gr.outputs.Image(type="pil", width=64, label="Output Image2"), gr.outputs.Image(type="pil", width=64, label="Output Image3"), gr.outputs.Image(type="pil", width=64, label="Output Image4")])
+#iface = gr.Interface(fn=greet, inputs=[gr.Textbox(label="Co-Retailing Business")], outputs=[gr.outputs.Image(type="pil", width=64, label="Output Image"), gr.outputs.Image(type="pil", width=64, label="Output Image2"), gr.outputs.Image(type="pil", width=64, label="Output Image3"), gr.outputs.Image(type="pil", width=64, label="Output Image4")])
+iface = gr.Interface(fn=greet, inputs=[gr.Textbox(label="Co-Retailing Business")], outputs=[gr.outputs.Image(type="pil", label="Output Image"), gr.outputs.Image(type="pil", label="Output Image2"), gr.outputs.Image(type="pil", label="Output Image3"), gr.outputs.Image(type="pil", label="Output Image4")])
 #iface = gr.Interface(fn=greet, inputs=[gr.Textbox(label="Co-Retailing Business")], outputs=[gr.Textbox()])
 iface.launch()