Update app.py

app.py

@@ -29,9 +29,13 @@ def predict(image,max_length=64, num_beams=4):
   caption_text = clean_text(tokenizer.decode(caption_ids))
   return caption_text 
 
-with gr.Blocks() as demo:
-
-    gr.HTML(
+print("------------------------- 5 -------------------------\n")
+input = gr.inputs.Image(label="Upload any Image", type = 'pil', optional=True)
+output = gr.outputs.Textbox(type="text",label="Captions")
+examples = ["example1.jpg"]
+print("------------------------- 6 -------------------------\n")
+# title = "Image to - Text"
+description = gr.HTML(
         """
         <div style="text-align: center; max-width: 1200px; margin: 20px auto;">
         <h1 style="font-weight: 900; font-size: 3rem; margin: 0rem">
@@ -43,14 +47,6 @@ with gr.Blocks() as demo:
         </h2>
         </div>
         """)
-    
-print("------------------------- 5 -------------------------\n")
-input = gr.inputs.Image(label="Upload any Image", type = 'pil', optional=True)
-output = gr.outputs.Textbox(type="text",label="Captions")
-examples = ["example1.jpg"]
-print("------------------------- 6 -------------------------\n")
-# title = "Image to - Text"
-description = "NTT Data"
 interface = gr.Interface(
             
         fn=predict,