Spaces:
Paused
Paused
update gradio Client for clip interrogator
Browse files
app.py
CHANGED
@@ -1,5 +1,6 @@
|
|
1 |
import gradio as gr
|
2 |
import os
|
|
|
3 |
|
4 |
title="Prompt Converter"
|
5 |
|
@@ -12,7 +13,7 @@ Stable Diffusion 2 uses OpenCLIP ViT-H model trained on LAION dataset so it know
|
|
12 |
"""
|
13 |
|
14 |
stable_diffusion = gr.load(name="spaces/runwayml/stable-diffusion-v1-5")
|
15 |
-
clip_interrogator_2 =
|
16 |
|
17 |
def get_images(prompt):
|
18 |
gallery_dir = stable_diffusion(prompt, fn_index=2)
|
@@ -20,7 +21,12 @@ def get_images(prompt):
|
|
20 |
return img_results[0]
|
21 |
|
22 |
def get_new_prompt(img, mode):
|
23 |
-
interrogate = clip_interrogator_2(
|
|
|
|
|
|
|
|
|
|
|
24 |
return interrogate
|
25 |
|
26 |
def infer(prompt, mode):
|
|
|
1 |
import gradio as gr
|
2 |
import os
|
3 |
+
from gradio_client import Client
|
4 |
|
5 |
title="Prompt Converter"
|
6 |
|
|
|
13 |
"""
|
14 |
|
15 |
stable_diffusion = gr.load(name="spaces/runwayml/stable-diffusion-v1-5")
|
16 |
+
clip_interrogator_2 = Client("https://fffiloni-clip-interrogator-2.hf.space/")
|
17 |
|
18 |
def get_images(prompt):
|
19 |
gallery_dir = stable_diffusion(prompt, fn_index=2)
|
|
|
21 |
return img_results[0]
|
22 |
|
23 |
def get_new_prompt(img, mode):
|
24 |
+
interrogate = clip_interrogator_2.predict(
|
25 |
+
img, # str (filepath or URL to image) in 'parameter_3' Image component
|
26 |
+
mode, # str in 'Select mode' Radio component
|
27 |
+
12, # int | float (numeric value between 2 and 24) in 'best mode max flavors' Slider component
|
28 |
+
api_name="/clipi2"
|
29 |
+
#interrogate = clip_interrogator_2(img, mode, 12, api_name="clipi2")
|
30 |
return interrogate
|
31 |
|
32 |
def infer(prompt, mode):
|