Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -61,18 +61,17 @@ def main():
|
|
61 |
|
62 |
# check if file has been uploaded
|
63 |
if uploaded_file is not None:
|
64 |
-
|
65 |
-
img = preprocess_image(uploaded_file)
|
66 |
-
|
67 |
# load the image
|
68 |
-
|
|
|
|
|
69 |
|
70 |
# context as prompt
|
71 |
prompt = generate_captions(img)
|
72 |
st.write("The Context is:", prompt)
|
73 |
|
74 |
# display the image
|
75 |
-
st.image(
|
76 |
|
77 |
# Generate button
|
78 |
if st.button("Generate"):
|
|
|
61 |
|
62 |
# check if file has been uploaded
|
63 |
if uploaded_file is not None:
|
|
|
|
|
|
|
64 |
# load the image
|
65 |
+
image = Image.open(uploaded_file).convert("RGB")
|
66 |
+
|
67 |
+
img = preprocess_image(image)
|
68 |
|
69 |
# context as prompt
|
70 |
prompt = generate_captions(img)
|
71 |
st.write("The Context is:", prompt)
|
72 |
|
73 |
# display the image
|
74 |
+
st.image(uploaded_file)
|
75 |
|
76 |
# Generate button
|
77 |
if st.button("Generate"):
|