Erasing-Concepts-In-Diffusion

Runtime error

RohitGandikota commited on Mar 21, 2023

Commit

397aa19

•

1 Parent(s): 8c1e922

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,8 @@ model_map = {'Car' : 'models/car.pt',
              'Kilian Eng' : 'models/kilianeng.pt',
              'Thomas Kinkade' : 'models/thomaskinkade.pt',
              'Tyler Edlin' : 'models/tyleredlin.pt',
-             'Kelly McKernan': '/models/kellymckernan.pt'}
 class Demo:

              'Kilian Eng' : 'models/kilianeng.pt',
              'Thomas Kinkade' : 'models/thomaskinkade.pt',
              'Tyler Edlin' : 'models/tyleredlin.pt',
+             'Kelly McKernan': '/models/kellymckernan.pt',
+             'Rembrandt': '/models/rembrandt.pt' }
 class Demo: