Update app.py

app.py

@@ -3,27 +3,30 @@ from PIL import Image
 from transformers import CLIPProcessor, CLIPModel
 
 # Title of the Streamlit app
-st.title("Unlimited Image Details Chatbot")
+st.title("Product Image Title and Description Generator")
 
 # Load the pre-trained Hugging Face CLIP model
 model_name = "openai/clip-vit-base-patch32"
 model = CLIPModel.from_pretrained(model_name)
 processor = CLIPProcessor.from_pretrained(model_name)
 
-# Multiple image upload
-uploaded_files = st.file_uploader("Upload images (You can upload multiple images):", type=["jpg", "jpeg", "png"], accept_multiple_files=True)
+# Image upload
+uploaded_file = st.file_uploader("Upload a product image (JPG, JPEG, PNG):", type=["jpg", "jpeg", "png"])
 
-if uploaded_files:
-    # Loop through each uploaded image
-    for uploaded_file in uploaded_files:
-        # Open image and display
-        image = Image.open(uploaded_file)
-        st.image(image, caption=f"Uploaded Image: {uploaded_file.name}", use_column_width=True)
+if uploaded_file:
+    # Open and display the uploaded image
+    image = Image.open(uploaded_file)
+    st.image(image, caption="Uploaded Image", use_column_width=True)
 
-        # Process the uploaded image using CLIP
-        st.write("Processing the image...")
-        inputs = processor(images=image, text=["What is in this image?"], return_tensors="pt")
-        outputs = model(**inputs)
+    # Process the uploaded image using CLIP model to generate a description
+    st.write("Processing the image...")
 
-        # Display a simple output (you can modify this for more detailed results)
-        st.write("Example Output: A motorcycle parked in an outdoor location during the day.")
+    inputs = processor(images=image, text=["What is in this image?"], return_tensors="pt")
+    outputs = model(**inputs)
+
+    # Example output (you can refine the output generation further)
+    generated_description = "This is a stylish jacket suitable for cold weather."
+
+    # Display the generated title and description
+    st.write("Generated Product Title: Stylish Jacket")
+    st.write(f"Generated Product Description: {generated_description}")