from diffusers import StableDiffusionXLPipeline
import torch
from transformers.pipelines.image_to_text import Image
from IPython.display import display
import gradio as gr


from diffusers.utils.import_utils import USE_SAFETENSORS
pipe = StableDiffusionXLPipeline.from_pretrained(
    "segmind/SSD-1B",
    tensor_dtype = torch.float16,
    USE_SAFETENSORS = True,
    variant = 'fp16'
)

prompt = 'most beutiful  woMEN young realistic high quality  '
neg_prompt = 'ugly, blurry, poor quality'

image = pipe(prompt = prompt , negative_prompt = neg_prompt).images[0]

display(image)


def greet(prompt):
    return "Hello " + prompt + "!!"

iface = gr.Interface(fn=greet, inputs="text", outputs="image")
iface.launch()