imagegenerator.py

!pip install gradio


import gradio as gr
import os
import requests

def text2image(prompt, key):

  API_URL = "https://api-inference.huggingface.co/models/Datou1111/shou_xin"
  headers = {"Authorization": "Bearer "+key}

  def query(payload):
    response = requests.post(API_URL, headers=headers, json=payload)
    return response.content
  image_bytes = query({
    "inputs": prompt,
  })
  # You can access the image with PIL.Image for example
  import io
  from PIL import Image
  image = Image.open(io.BytesIO(image_bytes))

  return image

#css = ".gradio-container {background: url(https://i.postimg.cc/RV0L7qcn/background-app.jpg)}"

demo = gr.Interface(
    fn=text2image,
    inputs=["text", gr.Textbox(label="password", type="password")],
    outputs=[gr.Image(label="Image Output")],
    #css=css,
    title = "ImageGen App"
)

demo.launch(share=True)