Update app.py

app.py

@@ -65,10 +65,15 @@ embedding_dim = 128
 hidden_dim = 64
 num_classes = len(labels)
 
+# Load the TensorFlow model
+model_tf = tflearn.DNN(tflearn.input_data(shape=[None, len(training[0])]))
+model_tf.load("MentalHealthChatBotmodel.tflearn")
+
+# Convert the TensorFlow model to a PyTorch model
 pytorch_model = PyTorchModel(vocab_size, embedding_dim, hidden_dim, num_classes)
 
-# Load weights from the tflearn model
-for layer_name, weights in zip(['fc1/kernel:0', 'fc1/bias:0', 'fc2/kernel:0', 'fc2/bias:0'], model.get_weights()):
+# Load weights from the TensorFlow model
+for layer_name, weights in zip(['fc1/kernel:0', 'fc1/bias:0', 'fc2/kernel:0', 'fc2/bias:0'], model_tf.get_weights()):
     pytorch_layer_name = layer_name.replace(':0', '')
     pytorch_model.state_dict()[pytorch_layer_name].copy_(torch.tensor(weights))