Update app.py

app.py

@@ -54,7 +54,6 @@ def sd2_inference(pipeline, prompts, params, seed = 42, num_inference_steps = 50
 HF_ACCESS_TOKEN = os.environ["HFAUTH"]
 
 # Load Model
-# - Reference: https://github.com/huggingface/diffusers/blob/main/README.md
 pipeline, params = FlaxStableDiffusionPipeline.from_pretrained(
     "CompVis/stable-diffusion-v1-4",
     use_auth_token = HF_ACCESS_TOKEN,
@@ -73,7 +72,6 @@ model = FlaxVisionEncoderDecoderModel.from_pretrained(loc)
 gen_kwargs = {"max_length": 16, "num_beams": 4}
 
 
-# This takes sometime when compiling the first time, but the subsequent inference will be much faster
 
 def generate(pixel_values):
     output_ids = model.generate(pixel_values, **gen_kwargs).sequences
@@ -95,11 +93,11 @@ def image2text(image):
 
 
 def text_to_image_and_image_to_text(text=None,image=None):
-    txt=None
+    txt=""
     img=None
     if image != None:
         txt=image2text(image)
-    if text !=None:
+    if text !="":
         images = sd2_inference(pipeline, [text], params, seed = 42, num_inference_steps = 5 )
         img = images[0]
     return img,txt
@@ -107,7 +105,7 @@ def text_to_image_and_image_to_text(text=None,image=None):
 
 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",value=None, label="Text to Encode to Image ",lines=1,optional=True),gr.Image(type="pil",value=None,label="Image to Decode to text",optional=True)],
+                             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: Text2Image2Text imformation transmiter",
                              description="⭐️The next generation of QR codes, an information sharing tool via images⭐️ Error rates are high & Image generation takes about 200 seconds.",