Update app.py

app.py

@@ -47,9 +47,12 @@ def main():
         # load the image
         image = Image.open(uploaded_file).convert("RGB")
         image = image.resize((224, 224))
+        image_array = np.array(image)
+        normalized_image = image_array / 255.0
+        reshaped_image = normalized_image.reshape((1, 224, 224, 3))
         
         # context as prompt
-        prompt = generate_captions(image)
+        prompt = generate_captions(reshaped_image)
         st.write("The Context is:", prompt)
 
         # display the image