Update app.py

app.py

@@ -59,31 +59,24 @@ pipeline, params = FlaxStableDiffusionPipeline.from_pretrained(
     dtype=jnp.bfloat16,
 )
 
-def text_to_image_and_image_to_text(text,image):
-    img=None
-    txt=None
-    if image "" != None:
+
+
+import gradio as gr
+
+def text_to_image_and_image_to_text(text=None,image=None):
+    if image != None:
+        txt=text
+    if text !=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_and_image_to_text,
-                             inputs=[gr.inputs.Textbox(placeholder="Enter the text to Encode to an image", label="Text "
-                                                                                                           "query",
-                                                      lines=1),
-                                    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",
-                             description="",
-                             theme="huggingface")
+                             inputs=[gr.inputs.Textbox(placeholder="Enter the text to Encode to an image", label="Text to Encode to Image ",lines=1,optional=True),gr.Image(type="pil",label="Image to Decode to text",optional=True)],
+                             outputs=[gr.outputs.Image(type="pil", label="Encoded Image"),gr.outputs.Textbox( label="Decoded Text")],
+                             title="T2I2T",
+                             description="T2I2T: Text2Image2Text imformation transmiter"
+                             )
     interFace.launch()