Yougen commited on
Commit
3792eb0
·
verified ·
1 Parent(s): 4142f6f

Update text2image.py

Browse files
Files changed (1) hide show
  1. text2image.py +2 -1
text2image.py CHANGED
@@ -9,7 +9,8 @@ from clip_retrieval.clip_client import ClipClient
9
 
10
  def text2image_gr():
11
  def clip_api(query_text='', return_n=8, model_name=clip_base, thumbnail="是"):
12
- client = ClipClient(url="http://9.135.121.52:1234//knn-service",
 
13
  indice_name="ltr_cover_index",
14
  aesthetic_weight=0,
15
  num_images=int(return_n))
 
9
 
10
  def text2image_gr():
11
  def clip_api(query_text='', return_n=8, model_name=clip_base, thumbnail="是"):
12
+ #client = ClipClient(url="http://9.135.121.52:1234//knn-service",
13
+ client = ClipClient(url="http://127.0.0.1:1234//knn-service",
14
  indice_name="ltr_cover_index",
15
  aesthetic_weight=0,
16
  num_images=int(return_n))