Update app.py

app.py

@@ -1,21 +1,13 @@
 import streamlit as st
 from PIL import Image
-from transformers import GPT2TokenizerFast, ViTImageProcessor, VisionEncoderDecoderModel,RobertaTokenizerFast
-
-
-import requests
-from PIL import Image
-
-from transformers import GPT2TokenizerFast, ViTImageProcessor, VisionEncoderDecoderModel
-
-
+from transformers import GPT2TokenizerFast, ViTImageProcessor, VisionEncoderDecoderModel,RobertaTokenizerFast, VisionEncoderDecoderModel
 
 #import torch
 #from transformers import BlipProcessor, BlipForConditionalGeneration
 
 def set_page_config():
     st.set_page_config(
-        page_title='Caption an Image', 
+        page_title='Caption an Cartoon Image', 
         page_icon=':camera:', 
         layout='wide',
     )
@@ -31,8 +23,7 @@ def initialize_model():
 def upload_image():
     return st.sidebar.file_uploader("Upload an image (we aren't storing anything)", type=["jpg", "jpeg", "png"])
 
-def resize_image(image, max_width):
-    
+def image_preprocess(image):
     image = image.resize((max_width, height))
     if image.mode == "L":
         image = image.convert("RGB")
@@ -53,28 +44,27 @@ def main():
 
     if uploaded_image is not None:
         image = Image.open(uploaded_image)
-        image = resize_image(image, max_width=300)
-
+        image = image_preprocess(image)
+        
         st.image(image, caption='Your image')
         
         with st.sidebar:
             st.divider() 
             if st.sidebar.button('Generate Caption'):
                 with st.spinner('Generating caption...'):
-                    #processor, model, device = initialize_model()
-                    #caption = generate_caption(processor, model, device, image)
-                    caption="im here man"
+                    image_processor, model,tokenizer, device = initialize_model()
+                    
+                    caption = generate_caption(image_processor, model, device, image)
+                    #caption="im here man"
                     st.header("Caption:")
                     st.markdown(f'**{caption}**')
 
 
-
 if __name__ == '__main__':
     main()
 
 
 st.markdown("""
     ---
-   You are looking at Finetuned image Caption model """)
-
+   You are looking at partial finetuned model , please JUDGE ME!!!  """)