Update app.py

app.py

@@ -8,7 +8,7 @@ import torch_utils
 
 # Load the pre-trained StyleGAN model
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-model_path = 'network.pkl'  # Place your .pkl in the same directory or update path
+model_path = 'dress_model.pkl'  # Place your .pkl in the same directory or update path
 
 # Load StyleGAN Generator
 with open(model_path, 'rb') as f: