Update HelperFunctions.py

HelperFunctions.py

@@ -85,8 +85,8 @@ stage2_generator = Stage2Generator(text_embedding_dim=768, img_size=64)
 stage1_generator.eval()
 stage2_generator.eval()
 device = 'cpu'
-stage1_generator.load_state_dict(torch.load('Weights/stage1Generator_weights.pth',map_location=device))
-stage2_generator.load_state_dict(torch.load('Weights/stage2Generator_weights_UPDATED.pth',map_location=device))
+stage1_generator.load_state_dict(torch.load('stage1Generator_weights.pth',map_location=device))
+stage2_generator.load_state_dict(torch.load('stage2Generator_weights_UPDATED.pth',map_location=device))
 print("Models loaded successfully")
     
 tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')