drod75's picture
Update app.py
ae8f30c verified
raw
history blame
364 Bytes
import streamlit as st
from transformers import pipeline
from diffusers import DiffusionPipeline
pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0")
st.title('Anime Character Generator')
prompt = st.text_input('Enter a prompt for your Anime character: ')
while prompt != '':
image = pipe(prompt).images[0]
st.image(image)