Update app.py

app.py

@@ -40,7 +40,7 @@ df = torch.load(f"{models_path}/files/identity_df.pt")
 weight_dimensions = torch.load(f"{models_path}/files/weight_dimensions.pt")
 pinverse = torch.load(f"{models_path}/files/pinverse_1000pc.pt", map_location=torch.device('cpu')).bfloat16().to(device.value)
 
-unet, vae, text_encoder, tokenizer, noise_scheduler = load_models(device)
+unet.value, vae.value, text_encoder.value, tokenizer.value, noise_scheduler.value = load_models(device.value)
 
 def sample_model():
     unet.value, _, _, _, _ = load_models(device)