RohitGandikota commited on
Commit
daebf8f
1 Parent(s): 397aa19

added new artists and message after training

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,8 +11,8 @@ model_map = {'Car' : 'models/car.pt',
11
  'Kilian Eng' : 'models/kilianeng.pt',
12
  'Thomas Kinkade' : 'models/thomaskinkade.pt',
13
  'Tyler Edlin' : 'models/tyleredlin.pt',
14
- 'Kelly McKernan': '/models/kellymckernan.pt',
15
- 'Rembrandt': '/models/rembrandt.pt' }
16
 
17
 
18
  class Demo:
 
11
  'Kilian Eng' : 'models/kilianeng.pt',
12
  'Thomas Kinkade' : 'models/thomaskinkade.pt',
13
  'Tyler Edlin' : 'models/tyleredlin.pt',
14
+ 'Kelly McKernan': 'models/kellymckernan.pt',
15
+ 'Rembrandt': 'models/rembrandt.pt' }
16
 
17
 
18
  class Demo: