from diffusers import StableDiffusionPipeline
import streamlit as st
import torch

generator = StableDiffusionPipeline.from_pretrained("AMAN-B/Demo-Dreambooth").to("cuda")

prompt = st.text_area("enter prompt")
if text:
	image = generator(prompt).images[0]
	st.json(image)