Spaces:
Build error
Build error
File size: 1,702 Bytes
d702758 ab70f27 d702758 ab70f27 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
import requests
from PIL import Image
import gradio as gr
from io import BytesIO
url = "https://stablediffusionapi.com/api/v3/text2img"
title = """<h2><center>Text to Image Generation with Stable Diffusion API</center></h2>"""
description = """#### Get the API key by signing up here [Stable Diffusion API](https://stablediffusionapi.com)."""
def get_image(key, prompt, inference_steps, filter):
payload = {
"key": key,
"prompt": prompt,
"negative_prompt": "((out of frame)), ((extra fingers)), mutated hands, ((poorly drawn hands)), ((poorly drawn face)), (((mutation))), (((deformed))), (((tiling))), ((naked)), ((tile)), ((fleshpile)), ((ugly)), (((abstract))), blurry, ((bad anatomy)), ((bad proportions)), ((extra limbs)), cloned face, (((skinny))), glitchy, ((extra breasts)), ((double torso)), ((extra arms)), ((extra hands)), ((mangled fingers)), ((missing breasts)), (missing lips), ((ugly face)), ((fat)), ((extra legs)), anime",
"width": "512",
"height": "512",
"samples": "1",
"num_inference_steps": inference_steps,"safety_checker": filter,"enhance_prompt": "yes","guidance_scale": 7.5}
headers = {}
response = requests.request("POST", url, headers=headers, data=payload)
url1 = str(json.loads(response.text)['output'][0])
r = requests.get(url1)
i = Image.open(BytesIO(r.content))
return i
demo = gr.Interface(fn=get_image,
inputs = [gr.Textbox(label="Enter API key"), gr.Textbox(label="Enter the Prompt"), gr.Number(label="Enter number of steps"),gr.Checkbox(label="Safety filter")],
outputs = gr.Image(type='pil'), title = title, description = description).launch(debug='True')
|