Spaces:
Runtime error
Runtime error
Commit
·
8e5e4bb
1
Parent(s):
8b67c6d
Update app.py
Browse files
app.py
CHANGED
@@ -42,7 +42,7 @@ uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "png"])
|
|
42 |
if uploaded_file is not None:
|
43 |
image = Image.open(uploaded_file)
|
44 |
st.image(image, caption='Uploaded Image.', clamp=True,use_column_width=True)
|
45 |
-
|
46 |
st.write("")
|
47 |
st.write("Generating caption...")
|
48 |
caption = generate_caption(image)
|
|
|
42 |
if uploaded_file is not None:
|
43 |
image = Image.open(uploaded_file)
|
44 |
st.image(image, caption='Uploaded Image.', clamp=True,use_column_width=True)
|
45 |
+
image = processor(images=image,return_tensors='pt').pixel_values
|
46 |
st.write("")
|
47 |
st.write("Generating caption...")
|
48 |
caption = generate_caption(image)
|