File size: 873 Bytes
7a4be7f 37ee6de 7a4be7f 37ee6de 7a4be7f 37ee6de 7a4be7f 37ee6de 7a4be7f 2455ada 7a4be7f 2455ada b636da7 7a4be7f |
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 |
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, # str (filepath or URL to image)
"ViT-L (best for Stable Diffusion 1.*)", # str (Option from: ['ViT-L (best for Stable Diffusion 1.*)'])
"best", # str in 'Mode' Radio component
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()
|