Update app.py

app.py

@@ -1,50 +1,24 @@
-import streamlit as st 
-from dotenv import load_dotenv
-import os 
-import openai
-from diffusers import StableDiffusionPipeline
-import torch
+import streamlit as st
+from PIL import Image
+from transformers import pipeline
 
-load_dotenv()
-openai.api_key = os.getenv("OPENAI_API_KEY")
+# Create an image captioning pipeline using Hugging Face Transformers
+image_captioner = pipeline("image-captioning")
 
-#function to generate AI based images using OpenAI Dall-E
-def generate_images_using_openai(text):
-    response = openai.Image.create(prompt= text, n=1, size="512x512")
-    image_url = response['data'][0]['url']
-    return image_url
+# Streamlit app header
+st.title("Hugging Face Transformer Image-to-Text App")
 
+# File uploader for image
+uploaded_image = st.file_uploader("Upload an image", type=["jpg", "png", "jpeg"])
 
-#function to generate AI based images using Huggingface Diffusers
-def generate_images_using_huggingface_diffusers(text):
-    pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
-    pipe = pipe.to("cuda")
-    prompt = text
-    image = pipe(prompt).images[0] 
-    return image
+if uploaded_image:
+    # Display the uploaded image
+    image = Image.open(uploaded_image)
+    st.image(image, caption="Uploaded Image", use_column_width=True)
 
-#Streamlit Code
-choice = st.sidebar.selectbox("Select your choice", ["Home", "DALL-E", "Huggingface Diffusers"])
+    # Perform image captioning when an image is uploaded
+    caption = image_captioner(uploaded_image)[0]["caption"]
 
-if choice == "Home":
-    st.title("AI Image Generation App")
-    with st.expander("About the App"):
-        st.write("This is a simple image generation app that uses AI to generates images from text prompt.")
-
-elif choice == "DALL-E":
-    st.subheader("Image generation using Open AI's DALL-E")
-    input_prompt = st.text_input("Enter your text prompt")
-    if input_prompt is not None:
-        if st.button("Generate Image"):
-            image_url = generate_images_using_openai(input_prompt)
-            st.image(image_url, caption="Generated by DALL-E")
-
-elif choice == "Huggingface Diffusers":
-    st.subheader("Image generation using Huggingface Diffusers")
-    input_prompt = st.text_input("Enter your text prompt")
-    if input_prompt is not None:
-        if st.button("Generate Image"):
-            image_output = generate_images_using_huggingface_diffusers(input_prompt)
-            st.info("Generating image.....")
-            st.success("Image Generated Successfully")
-            st.image(image_output, caption="Generated by Huggingface Diffusers")
+    # Display the generated caption
+    st.subheader("Generated Caption:")
+    st.write(caption)