hungchiayu commited on
Commit
8e5e4bb
·
1 Parent(s): 8b67c6d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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)