Update app.py

app.py

@@ -11,7 +11,7 @@ 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"
+bert_model_name = "anshupatel4298/bert-chatbot-model"
 
 # Define the model architecture that matches the original one
 def create_model():
@@ -55,8 +55,10 @@ if st.button("Send"):
     if user_input:
         if model_choice == "Basic Model":
             # Preprocess input for basic model
-            tokenized_input = tf.keras.preprocessing.text.Tokenizer().texts_to_sequences([user_input])
-            input_data = tf.keras.preprocessing.sequence.pad_sequences(tokenized_input, maxlen=MAX_SEQUENCE_LENGTH)
+            tokenizer = tf.keras.preprocessing.text.Tokenizer(num_words=1825)
+            tokenizer.fit_on_texts([user_input])
+            tokenized_input = tokenizer.texts_to_sequences([user_input])
+            input_data = tf.keras.preprocessing.sequence.pad_sequences(tokenized_input, maxlen=1825)  # Ensure padding to match the input shape
             prediction = basic_model.predict(input_data)
             response = np.argmax(prediction, axis=-1)[0]
         else: