Update app.py

app.py

@@ -157,24 +157,27 @@ def start():
 def main():
     st.title("Image Captioning with Transformer Models")
     image = st.file_uploader("Choose an image...", type=["jpg", "png", "jpeg"])
+    
     if image is not None:
-       
+        st.image(image, use_column_width=True)
         # image_bytes = uploaded_file.getvalue()
         # image = image_base64(image_bytes)
         # image = get_image(uploaded_file)
-        
-        accelerator = Accelerator()
-        device = accelerator.device
-        # device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-        config = get_config()
-        tokenizer = get_or_build_tokenizer(config)
-        model = get_model(config, len(tokenizer))
-        model = accelerator.prepare(model)
-        accelerator.load_state('models/')
-        # model = get_model(config, len(tokenizer))
-        # model.to(device)
-
-        text_output = process(model, image, tokenizer, device)
+        with st.empty():
+            st.write("Processing the image... Please wait.")
+            accelerator = Accelerator()
+            device = accelerator.device
+            # device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+            config = get_config()
+            tokenizer = get_or_build_tokenizer(config)
+            model = get_model(config, len(tokenizer))
+            model = accelerator.prepare(model)
+            accelerator.load_state('models/')
+            # model = get_model(config, len(tokenizer))
+            # model.to(device)
+            
+           
+            text_output = process(model, image, tokenizer, device)
         st.write(text_output)
 
 if __name__ == "__main__":