Update app.py

app.py

@@ -20,7 +20,7 @@ def generate_caption(encoded_image):
     }
 
     payload = {
-        "model": "llava-hf/llava-1.5-7b-hf",
+        "model": "Qwen/Qwen2-VL-7B-Instruct",
         "messages": [
             {
                 "role": "system",
@@ -37,11 +37,11 @@ def generate_caption(encoded_image):
                 "url": f"data:image/png;base64,{encoded_image}"
             }
         },
-        "temperature": 0.7
+        "temperature": 0
     }
 
     response = requests.post(API_URL, headers=headers, json=payload)
-    
+
     if response.status_code == 200:
         result = response.json()
         caption = result.get("choices", [{}])[0].get("message", {}).get("content", "No caption generated.")
@@ -53,35 +53,9 @@ def generate_caption(encoded_image):
 # Streamlit app
 def main():
     st.set_page_config(page_title="Image to Caption Converter", layout="centered", initial_sidebar_state="collapsed")
-    
-    # Gradient background style
-    st.markdown("""
-        <style>
-            body {
-                background: linear-gradient(135deg, #1e3c72, #2a5298);
-                color: white;
-                font-family: 'Arial', sans-serif;
-            }
-            .uploaded-image {
-                max-width: 100%;
-                border: 2px solid #ffffff;
-                border-radius: 10px;
-            }
-            .copy-button {
-                background-color: #ff8800;
-                color: white;
-                border: none;
-                border-radius: 5px;
-                padding: 10px 15px;
-                cursor: pointer;
-            }
-            .copy-button:hover {
-                background-color: #cc6b00;
-            }
-        </style>
-    """, unsafe_allow_html=True)
 
     st.title("🖼️ Image to Caption Converter")
+    st.markdown("Upload an image and let the AI generate a detailed caption for it.")
 
     uploaded_file = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
     if uploaded_file:
@@ -99,9 +73,7 @@ def main():
                     st.text_area("", caption, height=100, key="caption_area")
 
                     # Copy button
-                    if st.button("Copy to Clipboard"):
-                        st.code(caption, language="text")
-                        st.success("Caption copied to clipboard!")
+                    st.button("Copy to Clipboard", on_click=lambda: st.session_state.update({"caption": caption}))
 
 if __name__ == "__main__":
     main()