Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -19,6 +19,9 @@ def generate_caption(encoded_image):
|
|
19 |
"Content-Type": "application/json"
|
20 |
}
|
21 |
|
|
|
|
|
|
|
22 |
payload = {
|
23 |
"model": "Qwen/Qwen2-VL-7B-Instruct",
|
24 |
"messages": [
|
@@ -40,6 +43,7 @@ def generate_caption(encoded_image):
|
|
40 |
"temperature": 0
|
41 |
}
|
42 |
|
|
|
43 |
response = requests.post(API_URL, headers=headers, json=payload)
|
44 |
|
45 |
if response.status_code == 200:
|
@@ -58,6 +62,7 @@ def main():
|
|
58 |
st.markdown("Upload an image and let the AI generate a detailed caption for it.")
|
59 |
|
60 |
uploaded_file = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
|
|
|
61 |
if uploaded_file:
|
62 |
# Display the uploaded image
|
63 |
st.image(uploaded_file, caption="Uploaded Image", use_container_width=True)
|
@@ -66,13 +71,17 @@ def main():
|
|
66 |
if st.button("Generate Caption"):
|
67 |
with st.spinner("Generating caption..."):
|
68 |
encoded_image = convert_image_to_base64(uploaded_file)
|
|
|
|
|
|
|
|
|
69 |
caption = generate_caption(encoded_image)
|
70 |
|
71 |
if caption:
|
72 |
st.subheader("Generated Caption:")
|
73 |
st.text_area("", caption, height=100, key="caption_area")
|
74 |
|
75 |
-
# Copy button
|
76 |
st.button("Copy to Clipboard", on_click=lambda: st.session_state.update({"caption": caption}))
|
77 |
|
78 |
if __name__ == "__main__":
|
|
|
19 |
"Content-Type": "application/json"
|
20 |
}
|
21 |
|
22 |
+
# Print base64 to debug (make sure it's different for each image upload)
|
23 |
+
print(f"Base64 encoded image: {encoded_image[:100]}...") # Print only first 100 characters for brevity
|
24 |
+
|
25 |
payload = {
|
26 |
"model": "Qwen/Qwen2-VL-7B-Instruct",
|
27 |
"messages": [
|
|
|
43 |
"temperature": 0
|
44 |
}
|
45 |
|
46 |
+
# Send request to API
|
47 |
response = requests.post(API_URL, headers=headers, json=payload)
|
48 |
|
49 |
if response.status_code == 200:
|
|
|
62 |
st.markdown("Upload an image and let the AI generate a detailed caption for it.")
|
63 |
|
64 |
uploaded_file = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
|
65 |
+
|
66 |
if uploaded_file:
|
67 |
# Display the uploaded image
|
68 |
st.image(uploaded_file, caption="Uploaded Image", use_container_width=True)
|
|
|
71 |
if st.button("Generate Caption"):
|
72 |
with st.spinner("Generating caption..."):
|
73 |
encoded_image = convert_image_to_base64(uploaded_file)
|
74 |
+
|
75 |
+
# Debugging: print out the base64 string length to verify if it's changing
|
76 |
+
st.write(f"Encoded image length: {len(encoded_image)} characters")
|
77 |
+
|
78 |
caption = generate_caption(encoded_image)
|
79 |
|
80 |
if caption:
|
81 |
st.subheader("Generated Caption:")
|
82 |
st.text_area("", caption, height=100, key="caption_area")
|
83 |
|
84 |
+
# Copy button (if you want to implement this, use st.session_state)
|
85 |
st.button("Copy to Clipboard", on_click=lambda: st.session_state.update({"caption": caption}))
|
86 |
|
87 |
if __name__ == "__main__":
|