Spaces:

nielsr
/

CLIPSeg

Runtime error

nielsr HF staff commited on Nov 18, 2022

Commit

3159cac

1 Parent(s): 4111941

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,6 +8,7 @@ processor = CLIPSegProcessor.from_pretrained("CIDAS/clipseg-rd64-refined")
 model = CLIPSegForImageSegmentation.from_pretrained("CIDAS/clipseg-rd64-refined")
 def process_image(image, prompts):
   inputs = processor(text=prompts, images=[image] * len(prompts), padding="max_length", return_tensors="pt")
   # predict

 model = CLIPSegForImageSegmentation.from_pretrained("CIDAS/clipseg-rd64-refined")
 def process_image(image, prompts):
+  prompts = list(prompts)
   inputs = processor(text=prompts, images=[image] * len(prompts), padding="max_length", return_tensors="pt")
   # predict