Update app.py
Browse files
app.py
CHANGED
|
@@ -61,6 +61,7 @@ def search_similar(query_img):
|
|
| 61 |
results.sort(key=lambda x: x[1], reverse=True)
|
| 62 |
return [(f"dataset/{name}", f"Score: {score:.4f}") for name, score in results[:5]]
|
| 63 |
|
|
|
|
| 64 |
def add_image(name: str, image):
|
| 65 |
path = DATASET_DIR / f"{name}.jpg"
|
| 66 |
image.save(path)
|
|
|
|
| 61 |
results.sort(key=lambda x: x[1], reverse=True)
|
| 62 |
return [(f"dataset/{name}", f"Score: {score:.4f}") for name, score in results[:5]]
|
| 63 |
|
| 64 |
+
@spaces.GPU
|
| 65 |
def add_image(name: str, image):
|
| 66 |
path = DATASET_DIR / f"{name}.jpg"
|
| 67 |
image.save(path)
|