Spaces:
Runtime error
Runtime error
RohitGandikota
commited on
Commit
•
97b401e
1
Parent(s):
daebf8f
adding more models
Browse files
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 = {'
|
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',
|