Spaces:
Runtime error
Runtime error
RohitGandikota
commited on
Commit
•
48e9186
1
Parent(s):
46ccc0e
Update app.py
Browse files
app.py
CHANGED
@@ -8,6 +8,9 @@ from train import train
|
|
8 |
model_map = {
|
9 |
'Car' : 'models/car.pt',
|
10 |
'Van Gogh' : 'models/vangogh.pt',
|
|
|
|
|
|
|
11 |
}
|
12 |
|
13 |
|
|
|
8 |
model_map = {
|
9 |
'Car' : 'models/car.pt',
|
10 |
'Van Gogh' : 'models/vangogh.pt',
|
11 |
+
'Kilian Eng' : 'models/kilianeng.pt',
|
12 |
+
'Thomas Kinkade' : 'models/thomaskinkade.pt',
|
13 |
+
'Tyler Edlin' : 'models/tyleredlin.pt'
|
14 |
}
|
15 |
|
16 |
|