Update app.py

app.py

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