Update app.py

app.py

@@ -47,11 +47,11 @@ def main():
     image = Image.open(uploaded_file).convert("RGB")
 
     # context as prompt
-    prompt = generate_captions(image)
+    prompt = generate_captions(uploaded_file)
     st.write("The Context is:", prompt)
 
     # display the image
-    st.image(image, width=500, height=400)
+    st.image(image)
     
     # Generate button
     if st.button("Generate"):