Update app.py

app.py

@@ -2,16 +2,25 @@ import streamlit as st
 from transformers import BertTokenizer, TFBertForSequenceClassification
 import tensorflow as tf
 import numpy as np
-
+import requests
 import os
+
+# Ensure PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION is set to "python"
 os.environ["PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION"] = "python"
 
 # Paths to your models hosted on Hugging Face
 basic_model_url = "https://huggingface.co/anshupatel4298/bert-chatbot-model/resolve/main/basic_chatbot_model.h5"
+local_model_path = "basic_chatbot_model.h5"
 bert_model_name = "anshupatel4298/bert-chatbot-model/bert_model"
 
-# Load Basic Model
-basic_model = tf.keras.models.load_model(basic_model_url)
+# Download the Basic Model from the URL
+if not os.path.exists(local_model_path):
+    response = requests.get(basic_model_url)
+    with open(local_model_path, 'wb') as f:
+        f.write(response.content)
+
+# Load Basic Model from the local file
+basic_model = tf.keras.models.load_model(local_model_path)
 
 # Load BERT Model and Tokenizer
 bert_model = TFBertForSequenceClassification.from_pretrained(bert_model_name)