RohitGandikota commited on
Commit
97b401e
1 Parent(s): daebf8f

adding more models

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -5,9 +5,11 @@ from StableDiffuser import StableDiffuser
5
  from tqdm import tqdm
6
  from train import train
7
 
8
- model_map = {'Car' : 'models/car.pt',
9
- 'Van Gogh' : 'models/vangogh.pt',
10
  'Pablo Picasso': 'models/pablopicasso.pt',
 
 
 
11
  'Kilian Eng' : 'models/kilianeng.pt',
12
  'Thomas Kinkade' : 'models/thomaskinkade.pt',
13
  'Tyler Edlin' : 'models/tyleredlin.pt',
 
5
  from tqdm import tqdm
6
  from train import train
7
 
8
+ model_map = {'Van Gogh' : 'models/vangogh.pt',
 
9
  'Pablo Picasso': 'models/pablopicasso.pt',
10
+ 'Car' : 'models/car.pt',
11
+ 'Garbage Truck': 'models/garbagetruck.pt',
12
+ 'French Horn': 'models/frenchhorn.pt',
13
  'Kilian Eng' : 'models/kilianeng.pt',
14
  'Thomas Kinkade' : 'models/thomaskinkade.pt',
15
  'Tyler Edlin' : 'models/tyleredlin.pt',