Update app.py

app.py

@@ -250,7 +250,11 @@ class main():
         self.std.to(device)
         self.v.to(device)
         self.proj.to(device)
-        self.weights = torch.load("model.pt").to(device)  #.to(device)
+        self.weights = torch.load("model.pt").to(device)
+        self.young.to(device)
+        self.pointy.to(device)
+        self.wavy.to(device)
+        self.thick.to(device)
 
         network = LoRAw2w( self.weights.bfloat16(), self.mean.bfloat16(), self.std.bfloat16(), self.v[:, :1000].bfloat16(), 
                     self.unet,