Update app.py

app.py

@@ -21,10 +21,6 @@ from transformers import (
 clip = CLIPVisionModelWithProjection.from_pretrained("openai/clip-vit-base-patch32")
 processor = AutoProcessor.from_pretrained("openai/clip-vit-base-patch32")
 
-default_img = Image.open("gfg.png")
-default_img = processor(images=default_img,return_tensors='pt').pixel_values
-
-
 tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-small")
 t5 = T5ForConditionalGeneration.from_pretrained("google/flan-t5-small")
 
@@ -35,13 +31,13 @@ vt5.load_state_dict(torch.load('weights.bin',map_location=torch.device('cpu')))
 def generate_caption(image):
     # Your model code here
     caption = "This is a placeholder caption"
-    print(image)
+  
     caption = vt5.generate_caption(image)
     return caption
 
 st.title("Image Captioning App")
 #st.image(image.numpy().reshape(224,224,3), caption='Uploaded Image.', clamp=True,use_column_width=True)
-uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "png","jpeg"])
+uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "png"])
 
 if uploaded_file is not None:
     image = Image.open(uploaded_file)