t

app.py

@@ -163,12 +163,14 @@ def sample_ddim(n_sample, n=20):
 
 def greet(input):
 	samples, intermediate = sample_ddim(32, n=25)
-	response = im.fromarray(intermediate[24][0][1]).convert("RGB");
+	#response = im.fromarray(intermediate[24][0][1]).convert("RGB");
+    response = intermediate[24][0]
 	return response
 
 #iface = gr.Interface(fn=greet, inputs="text", outputs="text")
 #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", label="Output Image"))
+#iface = gr.Interface(fn=greet, inputs=[gr.Textbox(label="Co-Retailing Business")], outputs=gr.outputs.Image(type="pil", label="Output Image"))
+iface = gr.Interface(fn=greet, inputs=[gr.Textbox(label="Co-Retailing Business")], outputs="Text"))
 iface.launch()