Spaces:
Configuration error
Configuration error
import gradio as gr | |
import diffusers | |
import streamlit as st | |
device = "cpu" | |
from diffusers import StableDiffusionPipeline | |
pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", revision = "fp16", use_auth_token = st.secrets["USER_TOKEN"]) | |
pipe = pipe.to("cpu") | |
from PIL import Image | |
import torch | |
def StableDiffusionPipeline (prompt, Guide, iSteps, seed): | |
generator = torch.Generator("cpu").manual_seed(seed) | |
image = pipe(prompt, num_inference_steps = iSteps, guidence_scale = Guide).images[0] | |
return image | |
iface = gr.Interface(fn = StableDiffusionPipeline, inputs = [ | |
gr.Textbox(label = 'Prompt Input Text'), | |
gr.Slider(2, 15, value = 7, label = 'Guidence Scale'), | |
gr.Slider(10, 100, value = 25, step = 1, label = 'Number of Iterations'), | |
gr.Slider( | |
label = "Seed", | |
minimum = 0, | |
maximum = 2147483647, | |
step = 1, | |
randomize = True) | |
], | |
outputs = 'image') | |
iface.launch() |