fffiloni commited on
Commit
26359f2
·
1 Parent(s): 25135be

update gradio Client for clip interrogator

Browse files
Files changed (1) hide show
  1. app.py +8 -2
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 = gr.load(name="spaces/fffiloni/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(img, mode, 12, api_name="clipi2")
 
 
 
 
 
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):