new update

app.py

@@ -4,18 +4,17 @@ import pickle
 import numpy as np
 from huggingface_hub import hf_hub_download
 import os
-
+from tensorflow.keras.models import load_model
 
 def load_model():
     repo_id = 'Preethamreddy799/NLP_MODEL'
-    filename = 'model_test_steps.pkl'
+    filename = 'model_test_steps.h5'  # Assuming the model is in HDF5 format
 
-    # Get model URL and download
+    # Download the model from Hugging Face repository
     cached_model_path = hf_hub_download(repo_id=repo_id, filename=filename)
 
     # Load the model from the cached path
-    with open(cached_model_path, 'rb') as file:
-        model = pickle.load(file)
+    model = load_model(cached_model_path)
 
     print(f"Model loaded successfully from {cached_model_path}")
     return model

test.py

@@ -0,0 +1,10 @@
+import pickle
+
+downloads_path = '/Users/preethamreddygollapalli/Downloads/'
+
+# Load the saved models
+with open(downloads_path + 'model_test_steps.pkl', 'rb') as file:
+    model = pickle.load(file)
+    print(type(model))
+    print(model)
+