File size: 1,399 Bytes
18a5673 65fe587 18a5673 22dcbce 18a5673 7044dfd 18a5673 22dcbce ecd93f7 22dcbce 65fe587 22dcbce ae12cbe 22dcbce 5550780 22dcbce 18a5673 65fe587 22dcbce ae12cbe 65fe587 ae12cbe 65fe587 22dcbce fdc196f |
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 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
import gradio as gr
import requests
import json
import time
url = "https://nexra.aryahcr.cc/api/image/complements"
headers = {"Content-Type": "application/json"}
def generate_image(prompt, model):
data = {"prompt": prompt, "model": model}
response = requests.post(url, headers=headers, data=json.dumps(data))
if response.status_code == 200:
response_data = response.json()
image_id = response_data.get("id")
while True:
status_response = requests.get(f"{url}/{image_id}")
if status_response.status_code == 200:
status_data = status_response.json()
status = status_data.get("status")
if status == "completed":
images = status_data.get("images")
if images:
return images[0]
return no_nsfw_image_path
elif status in ["error", "not_found"]:
return no_nsfw_image_path
time.sleep(1)
return no_nsfw_image_path
iface = gr.Interface(
fn=generate_image,
inputs=[gr.Textbox(label="Enter prompt"), gr.Radio(["dalle2"], label="Select Model")],
outputs=gr.Image(type="filepath"),
title="DALLE2 Generation",
description="DALLE-2 Generation from Nextra API, NOTE: DALL-E 3 is not supported yet."
)
iface.launch(share=True)
|