Update app.py

app.py

@@ -101,6 +101,24 @@ def main():
         <p>Upload an image and get AI-generated captions instantly!</p>
     </div>
     """, unsafe_allow_html=True)
+    
+    # Sidebar
+    with st.sidebar:
+        st.header("🔧 Settings")
+        st.markdown("### Model Information")
+        st.info("Using **BLIP-2** (Salesforce/blip2-opt-2.7b)")
+        
+        st.markdown("### About")
+        st.markdown("""
+        This app uses the **BLIP-2** model to generate natural language descriptions of images.
+        
+        **Features:**
+        - 🖼️ Upload any image format
+        - 🤖 AI-powered captioning  
+        - ⚡ Fast inference
+        - 🎯 Optional custom prompts
+        """)
+    
     # Main content
     col1, col2 = st.columns([1, 1])
     
@@ -143,7 +161,7 @@ def main():
                         
                         # Generate caption
                         caption = generate_caption(
-                            image, processor, model, device, custom_prompt
+                            image, processor, model, device
                         )
                         
                         end_time = time.time()
@@ -176,7 +194,7 @@ def main():
     st.markdown("---")
     st.markdown("""
     <div style="text-align: center; color: #666;">
-        <p>Built using <strong>Streamlit</strong> and <strong>Hugging Face Transformers</strong></p>
+        <p>Built with ❤️ using <strong>Streamlit</strong> and <strong>Hugging Face Transformers</strong></p>
         <p>Powered by <strong>BLIP-2</strong> - Bootstrapping Language-Image Pre-training</p>
     </div>
     """, unsafe_allow_html=True)