import gradio as gr
import requests
import io
from PIL import Image
from transformers import pipeline

title = "Fine Tuned SD Model - Authoral stylization"
description = "Generate images trained in an authoral illustration model."
article = "<p style='text-align: center'><a href='https://news.machinelearning.sg/posts/beautiful_profile_pics_remove_background_image_with_deeplabv3/'>Blog</a> | <a href='https://github.com/eugenesiow/practical-ml'>Github Repo</a></p>"

def generate_image(prompt):
    model = pipeline("image-to-image", model="Cacau/heart-of-painting")
    return model(prompt)[0]

inputs = gr.inputs.Textbox(label="Prompt", placeholder="Your prompt here...")
output = gr.outputs.Image(label="Generated Image")

gr.Interface(generate_image, inputs, output, title=title, description=description, article=article).launch()