Update app.py

app.py

@@ -174,7 +174,7 @@ out_dropout_rate = 0.5;
 
 discriminator = Discriminator(input_size=hidden_size*2, hidden_sizes=hidden_levels_d, num_labels=len(label_list), dropout_rate=out_dropout_rate).to(device)
 discriminator_weights = ('data/datasets/discriminator_weights.pth')
-discriminator.load_state_dict(torch.load(discriminator_weights))
+discriminator.load_state_dict(torch.load(discriminator_weights,map_location=torch.device('cpu')))
 discriminator.eval()