%matplotlib inline

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

!pip install transformers
!pip install diffusers["torch"]
!pip install sentencepiece

import torch

from diffusers import StableDiffusionPipeline


pipe = StableDiffusionPipeline.from_pretrained("andite/anything-v4.0", torch_dtype=torch.float16)
pipe = pipe.to("cuda")

prompt = "a cute girl standing in front of chalkboard"

image = pipe(prompt).images[0]

image

!pip install gradio

import gradio as gr

def text2image(prompt):
    image = pipe(prompt).images[0]
    return image

inputs = gr.Textbox(label="輸入你的咒語")
outputs = gr.Image(label="生成圖")

iface = gr.Interface(text2image,
                     inputs=inputs,
                     outputs=outputs,
                     title="文字生圖 AI",
                     description="請輸入一段英文咒語 (prompt), 我會根據你的咒語畫出一張圖!")

iface.launch(share=True)