kusumakar commited on
Commit
7c2ca5d
·
1 Parent(s): bb931bd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
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
- #image = Image.open(uploaded_file).convert("RGB")
 
 
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(img)
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"):