ImageGenerate / app.py
Rakesh443's picture
Update app.py
771240c verified
raw
history blame
1.25 kB
import streamlit as st
import requests
import io
from PIL import Image
import os
import cv2
import numpy as np
# Create a text input
api_key = os.getenv("ImageGenerating")
API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
headers = {"Authorization": "f'Bearer {api_key}"}
user_input = st.text_input("Enter your text here:")
# Process the input (you can replace this with your own logic)
processed_output = user_input.upper()
# Display the processed output
# st.write(f"Processed output: {processed_output}")
def query(payload):
response = requests.post(API_URL, headers=headers, json=payload)
return response.content
image_bytes = query({
"inputs": user_input,
})
def bytes_to_image_opencv(image_bytes):
np_arr = np.frombuffer(image_bytes, np.uint8)
image = cv2.imdecode(np_arr, cv2.IMREAD_COLOR)
return image
result_image_opencv = bytes_to_image_opencv(image_bytes)
st.image(result_image_opencv,caption="image")
# image = Image.open(io.BytesIO(image_bytes))
# st.image(image, caption=None, width=None, use_column_width=None, clamp=False, channels="RGB", output_format="PNG")
# st.image(image=image,caption="image")
# st.image(image)
# st.image(image : image,caption=image)