Spaces:

ifmain
/

Object_Remove

Running on Zero

Mike Afton commited on Oct 7, 2023

Commit

64712db

•

1 Parent(s): 173fbaa

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -108,7 +108,7 @@ def getDescript(image_captioner,img1):
     return caption
 def rmGPT(caption,remove_class):
-    arstr=list(caption.split(' '))
     popular=get_most_similar_string(remove_class,arstr)
     ind=arstr.index(popular)
     new=[]

     return caption
 def rmGPT(caption,remove_class):
+    arstr=caption.split(' ')
     popular=get_most_similar_string(remove_class,arstr)
     ind=arstr.index(popular)
     new=[]