import streamlit as st
from transformers import pipeline

st.header('IMA-GEN', divider='red')
st.text('generate images from <text>')
prompt = st.text_input('prompt')
initImage = st.file_uploader('init image')

steps = st.number_input("steps", value=None, placeholder="Type a number")
st.write('current steps ', steps)