interior_designer / integrations.py
lemonteaa's picture
First attempt
df511cb
raw
history blame contribute delete
614 Bytes
from gradio_client import Client
import random
import time
def public_sdxl_gen(prompt : str, neg_prompt : str, progress : callable, n, poll_time):
seed = random.randrange(1000000000, 1999999999)
client = Client("https://victor-sdxl-0-9.hf.space/")
job = client.submit(prompt, neg_prompt, 10, 50, seed, api_name="/predict")
while not job.done():
job_status = job.status()
progress_str = "{r}/{q}, eta = {eta}".format(r=job_status.rank,q=job_status.queue_size,eta=job_status.eta)
progress(n, desc=progress_str)
time.sleep(poll_time)
return (seed, job.result())