Update app.py

app.py

@@ -76,8 +76,7 @@ pytorch_model = PyTorchModel(vocab_size, embedding_dim, hidden_dim, num_classes)
 layer_names = ['fc1/kernel', 'fc1/bias', 'fc2/kernel', 'fc2/bias']
 
 for layer_name in layer_names:
-    weight_tensor = model_tf.layers[layer_name].W.eval(session=model_tf.trainer.session)
-    bias_tensor = model_tf.layers[layer_name].b.eval(session=model_tf.trainer.session)
+    weight_tensor, bias_tensor = model_tf.get_weights(layer_name)
     pytorch_layer_name = layer_name.replace('/', '_')
     pytorch_model.state_dict()[f"{pytorch_layer_name}_weight"].copy_(torch.tensor(weight_tensor))
     pytorch_model.state_dict()[f"{pytorch_layer_name}_bias"].copy_(torch.tensor(bias_tensor))