Update app.py

app.py

@@ -76,9 +76,9 @@ 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:
-    weights = model_tf.get_weights(layer_name)
+    weight_tensor = getattr(model_tf, layer_name)
     pytorch_layer_name = layer_name.replace('/', '_')
-    pytorch_model.state_dict()[pytorch_layer_name].copy_(torch.tensor(weights))
+    pytorch_model.state_dict()[pytorch_layer_name].copy_(torch.tensor(weight_tensor.eval(session=model_tf.trainer.session)))
 
 # Move the model to the CPU
 pytorch_model.cpu()