Update app.py

app.py

@@ -170,6 +170,7 @@ num_hidden_layers_d = 2;
 hidden_levels_d = [hidden_size for i in range(0, num_hidden_layers_d)]
 label_list = ["1", "0"]
 label_list.append('UNL')
+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')