Update app.py

app.py

@@ -2,11 +2,11 @@ import streamlit as st
 from transformers import pipeline
 from PIL import Image
 
-# Page setup
-st.set_page_config(page_title="TinyLLaVA - Snapshot", layout="centered")
-st.title("📷 TinyLLaVA - Snapshot Q&A (Hugging Face Spaces)")
+# Page config
+st.set_page_config(page_title="TinyLLaVA Snapshot Q&A", layout="centered")
+st.title("📸 TinyLLaVA - Snapshot Q&A (Hugging Face Spaces)")
 
-# Load TinyLLaVA
+# Load TinyLLaVA pipeline
 pipe = pipeline(
     task="image-to-text",
     model="bczhou/tiny-llava-v1-hf",
@@ -14,21 +14,19 @@ pipe = pipeline(
     device_map="cpu"
 )
 
-# Camera input
-captured_image = st.camera_input("📸 Take a photo")
+# Always visible camera input
+image = st.camera_input("📷 Take a snapshot")
 
-# Prompt
-prompt = st.text_input("💬 What do you want to ask about this image?", value="Describe this scene.")
+# Prompt text input
+prompt = st.text_input("💬 Your question:", value="Describe this scene.")
 
-# Run when both available
-if captured_image and prompt:
-    img = Image.open(captured_image).convert("RGB")
+# If both present, run inference
+if image is not None and prompt:
+    img = Image.open(image).convert("RGB")
     st.image(img, caption="Your Snapshot", use_column_width=True)
 
-    # Format query
     query = f"USER: <image>\n{prompt}\nASSISTANT:"
-
-    with st.spinner("TinyLLaVA is thinking..."):
+    with st.spinner("TinyLLaVA is generating..."):
         result = pipe(query, img)
         answer = result[0]["generated_text"]