RohitGandikota commited on
Commit
48e9186
1 Parent(s): 46ccc0e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
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