Update model_loading.py

model_loading.py

@@ -1,8 +1,9 @@
-from transformers import TFBertForSequenceClassification, BertTokenizerFast
+import os
+import tensorflow as tf
+from dotenv import load_dotenv
 
-# Define the model name
-model_name = "Erfan11/Neuracraft"
+load_dotenv()
+model_path = os.getenv('MODEL_PATH')
 
-# Load the TensorFlow model and tokenizer
-model = TFBertForSequenceClassification.from_pretrained(model_name, use_auth_token="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")
-tokenizer = BertTokenizerFast.from_pretrained(model_name, use_auth_token="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")
+def load_model():
+    return tf.keras.models.load_model(model_path)