Spaces:
Build error
Build error
mnh
commited on
Commit
·
1764399
1
Parent(s):
c79b019
test
Browse files
app.py
CHANGED
|
@@ -7,7 +7,7 @@ model, preprocess = clip.load("ViT-B/32", device=device)
|
|
| 7 |
|
| 8 |
|
| 9 |
def predict(image):
|
| 10 |
-
labels = "Japanese, Chinese, Roman, Greek, Etruscan, Scandinavian, Celtic, Medieval, Victorian, Neoclassic, Romanticism, Art Nouveau, Art deco
|
| 11 |
labels = labels.split(',')
|
| 12 |
|
| 13 |
image = preprocess(image).unsqueeze(0).to(device)
|
|
|
|
| 7 |
|
| 8 |
|
| 9 |
def predict(image):
|
| 10 |
+
labels = "Japanese, Chinese, Roman, Greek, Etruscan, Scandinavian, Celtic, Medieval, Victorian, Neoclassic, Romanticism, Art Nouveau, Art deco"
|
| 11 |
labels = labels.split(',')
|
| 12 |
|
| 13 |
image = preprocess(image).unsqueeze(0).to(device)
|