Update app.py

app.py

@@ -59,21 +59,28 @@ pipeline, params = FlaxStableDiffusionPipeline.from_pretrained(
     dtype=jnp.bfloat16,
 )
 
-def text_to_image(text):
-    images = sd2_inference(pipeline, [text], params, seed = 42, num_inference_steps = 5 )
-    img = images[0]
-    return img
+def text_to_image_and_image_to_text(text,image):
+    img=None
+    txt=None
+    if image "" != None:
+        images = sd2_inference(pipeline, [text], params, seed = 42, num_inference_steps = 5 )
+        img = images[0]
+        img = image
+    if text !=None:
+        txt=text
+    return img,txt
 
 examples = ["apple",
             "banana",
             "chocolate"]
 
 if __name__ == '__main__':
-    interFace = gr.Interface(fn=text_to_image,
-                             inputs=gr.inputs.Textbox(placeholder="Enter the text to Encode to an image", label="Text "
+    interFace = gr.Interface(fn=text_to_image_and_image_to_text,
+                             inputs=[gr.inputs.Textbox(placeholder="Enter the text to Encode to an image", label="Text "
                                                                                                            "query",
                                                       lines=1),
-                             outputs=gr.outputs.Image(type="auto", label="Generated Image"),
+                                    gr.Image(type="pil")]
+                             outputs=[gr.outputs.Image(type="auto", label="Generated Image"),gr.outputs.Textbox(placeholder="Decoded Text")],
                              verbose=True,
                              examples=examples,
                              title="Generate Image from Text",