Update app.py

app.py

@@ -2,39 +2,42 @@ import streamlit as st
 from transformers import pipeline
 from PIL import Image
 
-# Setup
-st.set_page_config(page_title="TinyLLaVA Snapshot Q&A", layout="centered")
-st.title("📸 TinyLLaVA — Snapshot Q&A (Spaces Safe)")
+# Page config
+st.set_page_config(page_title="TinyLLaVA Snapshot", layout="centered")
+st.title("📸 TinyLLaVA — Snapshot Q&A (Hugging Face Spaces Safe)")
 
-# Load model
+# Load TinyLLaVA pipeline
+st.info("Loading TinyLLaVA... Please wait ⏳")
 pipe = pipeline(
     task="image-to-text",
     model="bczhou/tiny-llava-v1-hf",
     trust_remote_code=True,
     device_map="cpu"
 )
+st.success("✅ Model loaded!")
 
-# 👇 Clear info so the widget tree is stable
-st.markdown("Use your **webcam** OR upload a file:")
+# Widgets
+st.markdown("**1️⃣ Take a snapshot OR upload an image**")
+image = st.camera_input("📷 Click 'Take Photo' below:")
 
-# Try webcam first
-image = st.camera_input("📷 Take a snapshot")
-
-# Fallback uploader
 if not image:
-    image = st.file_uploader("📂 Or upload an image", type=["jpg", "png", "jpeg"])
+    image = st.file_uploader("📂 Or upload an image file", type=["jpg", "png", "jpeg"])
+
+st.markdown("**2️⃣ Enter your question:**")
+prompt = st.text_input("💬", value="Describe this scene.")
 
-# Prompt
-prompt = st.text_input("💬 Your question:", value="Describe this scene.")
+# Debug info
+st.write(f"📸 Image uploaded: {'Yes' if image else 'No'}")
+st.write(f"💬 Prompt: {prompt}")
 
-# Run TinyLLaVA
-if image is not None and prompt:
+# If both present
+if image and prompt:
     img = Image.open(image).convert("RGB")
     st.image(img, caption="Your Image", use_column_width=True)
 
     query = f"USER: <image>\n{prompt}\nASSISTANT:"
 
-    with st.spinner("Generating..."):
+    with st.spinner("TinyLLaVA is generating..."):
         result = pipe(query, img)
         answer = result[0]["generated_text"]