Update app.py

app.py

@@ -384,6 +384,7 @@ def register(state, drawpad):
 
 @spaces.GPU(duration=120)
 def run(state, drawpad):
+    state.model = model
     state.model.device = torch.device('cuda')
     state.model.reset_seed(state.model.generator, opt.seed)
     state.model.reset_latent()