File size: 661 Bytes
5fa778a 3c75a18 5fa778a 3c75a18 5fa778a 3c75a18 5fa778a 3c75a18 5fa778a 3c75a18 5fa778a 3c75a18 5fa778a 3c75a18 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | from diffusers import StableDiffusionPipeline
import torch
import gradio as gr
model_name = 'nitrosocke/Arcane-Diffusion' # Choose one model from your list
def TextToImage(Prompt, model_name):
pipe = StableDiffusionPipeline.from_pretrained(model_name, torch_dtype=torch.float16)
pipe = pipe.to("cpu")
prompt = Prompt
image = pipe(prompt).images[0]
return image
interface = gr.Interface(fn=TextToImage,
inputs=["text", gr.inputs.Text()], # Use gr.inputs.Text() here
outputs="image",
title='Text to Image',
live=True)
interface.launch()
|