Spaces:
Sleeping
Sleeping
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)
|