nerusskyhigh commited on
Commit
fd9bf1e
1 Parent(s): 88b2d3a

add missing function

Browse files
Files changed (1) hide show
  1. app.py +12 -0
app.py CHANGED
@@ -1,6 +1,18 @@
1
  from fastai.vision.all import *
2
  print("Import OK")
3
 
 
 
 
 
 
 
 
 
 
 
 
 
4
  learn = load_learner('drawingstyle.pkl')
5
  print("Loader OK")
6
 
 
1
  from fastai.vision.all import *
2
  print("Import OK")
3
 
4
+
5
+ import string
6
+
7
+ def getClassName(fileName):
8
+ fileName = fileName[:-4] # remove extension .jpg
9
+ names = fileName.split('_')
10
+ artist = ''.join(name+" " for name in names if name[0] in string.ascii_letters)
11
+
12
+ return artist[:-1] #Remove last space
13
+
14
+
15
+
16
  learn = load_learner('drawingstyle.pkl')
17
  print("Loader OK")
18