|
import gradio_client |
|
from gradio_client import Client, file |
|
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("fffiloni/CLIP-Interrogator-2") |
|
client.view_api() |
|
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(type= 'filepath'),gr.Image(type= 'filepath')) |
|
demo.launch() |
|
|