fffiloni commited on
Commit
bc54443
·
verified ·
1 Parent(s): 0f6487e

update clip interrogator call

Browse files
Files changed (1) hide show
  1. app.py +10 -6
app.py CHANGED
@@ -2,7 +2,10 @@ import gradio as gr
2
  import re
3
  import os
4
  hf_token = os.environ.get('HF_TOKEN')
5
- from gradio_client import Client
 
 
 
6
 
7
  from transformers import AutoTokenizer, AutoModelForCausalLM
8
 
@@ -48,11 +51,12 @@ def get_text_after_colon(input_text):
48
 
49
  def infer(image_input, audience):
50
  gr.Info('Calling CLIP Interrogator ...')
51
- clipi_result = clipi_client.predict(
52
- image_input, # str (filepath or URL to image) in 'parameter_3' Image component
53
- "best", # str in 'Select mode' Radio component
54
- 4, # int | float (numeric value between 2 and 24) in 'best mode max flavors' Slider component
55
- api_name="/clipi2"
 
56
  )
57
  print(clipi_result)
58
 
 
2
  import re
3
  import os
4
  hf_token = os.environ.get('HF_TOKEN')
5
+
6
+ from gradio_client import Client, handle_file
7
+
8
+ clipi_client = Client("fffiloni/CLIP-Interrogator-2")
9
 
10
  from transformers import AutoTokenizer, AutoModelForCausalLM
11
 
 
51
 
52
  def infer(image_input, audience):
53
  gr.Info('Calling CLIP Interrogator ...')
54
+
55
+ clipi_result = client.predict(
56
+ image=handle_file(image_input),
57
+ mode="best",
58
+ best_max_flavors=4,
59
+ api_name="/clipi2"
60
  )
61
  print(clipi_result)
62