Spaces:
Build error
Build error
nerusskyhigh
commited on
Commit
•
fd9bf1e
1
Parent(s):
88b2d3a
add missing function
Browse files
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 |
|