Update app.py

app.py

@@ -50,9 +50,10 @@ def main():
         image_array = np.array(image)
         normalized_image = image_array / 255.0
         reshaped_image = normalized_image.reshape((1, 224, 224, 3))
+        image_tensor = torch.from_numpy(normalized_image).permute(2, 0, 1).unsqueeze(0).float()
         
         # context as prompt
-        prompt = generate_captions(reshaped_image)
+        prompt = generate_captions(image_tensor)
         st.write("The Context is:", prompt)
 
         # display the image