Update app.py

app.py

@@ -184,6 +184,7 @@ class main():
         self.unet.to(device)
         self.text_encoder.to(device)
         self.vae.to(device)
+        self.network.to(device)