mnh commited on
Commit
1764399
·
1 Parent(s): c79b019
Files changed (1) hide show
  1. app.py +1 -1
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, Cyberpunk "
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)