Spaces:

ancebuc
/

CLIPSegmentation

Sleeping

ancebuc commited on May 13

Commit

c6cb1c5

•

1 Parent(s): cf253ac

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -31,7 +31,7 @@ def segment(img, clases):
     outputs = model(**inputs)
     preds = outputs.logits.unsqueeze(1)
-    return "Hello " + prompts + "!!"
 demo = gr.Interface(fn=segment, inputs=["image","text"], outputs="text")
 demo.launch()

     outputs = model(**inputs)
     preds = outputs.logits.unsqueeze(1)
+    return "Hello " + prompts[0] + "!!"
 demo = gr.Interface(fn=segment, inputs=["image","text"], outputs="text")
 demo.launch()