Update app.py
Browse files
app.py
CHANGED
|
@@ -7,7 +7,7 @@ model = AraClip.from_pretrained("Arabic-Clip/araclip")
|
|
| 7 |
def search(labels, image):
|
| 8 |
# process labels
|
| 9 |
labels = labels.split(",")
|
| 10 |
-
labels = [item.strip() for item in
|
| 11 |
|
| 12 |
# embed data
|
| 13 |
image_features = model.embed(image=image)
|
|
|
|
| 7 |
def search(labels, image):
|
| 8 |
# process labels
|
| 9 |
labels = labels.split(",")
|
| 10 |
+
labels = [item.strip() for item in labels if item != ""]
|
| 11 |
|
| 12 |
# embed data
|
| 13 |
image_features = model.embed(image=image)
|