|
from gradio_client import Client |
|
import numpy as np |
|
import gradio as gr |
|
def generate_img(prompt): |
|
client = Client("ameerazam08/SDXS-GPU-Demo") |
|
client.view_api() |
|
result = client.predict( |
|
prompt=prompt, |
|
api_name="/generate_image" |
|
) |
|
return result |
|
|
|
|
|
|
|
def interrogate(img): |
|
from gradio_client import Client |
|
|
|
client = Client("https://pharmapsychotic-clip-interrogator.hf.space/") |
|
client.view_api() |
|
result = client.predict( |
|
img, |
|
"ViT-L (best for Stable Diffusion 1.*)", |
|
"best", |
|
fn_index=3 |
|
) |
|
return result |
|
def rountrip(img): |
|
prompt=interrogate(img) |
|
return generate_img(prompt) |
|
|
|
demo = gr.Interface(rountrip, gr.Image(),gr.Image()) |
|
demo.launch() |
|
|