Update app.py

app.py

@@ -45,7 +45,7 @@ def main():
     # check if file has been uploaded
     if uploaded_file is not None:
         # load the image
-        image = np.array(Image.open(uploaded_file).resize((224, 224))).transpose(2, 0, 1)
+        image = Image.open(uploaded_file).convert("RGB")
         
         # context as prompt
         prompt = generate_captions(image)