Update app.py

app.py

@@ -13,10 +13,11 @@ basic_model_url = "https://huggingface.co/anshupatel4298/bert-chatbot-model/reso
 local_model_path = "basic_chatbot_model.h5"
 bert_model_name = "anshupatel4298/bert-chatbot-model/bert_model"
 
-# Define the model architecture
+# Define the model architecture that matches the original one
 def create_model():
     model = tf.keras.Sequential([
-        tf.keras.layers.Dense(128, activation='relu', input_shape=(100,)),  # Adjust input shape accordingly
+        tf.keras.layers.InputLayer(input_shape=(1825,)),  # Adjust input shape to match the original model
+        tf.keras.layers.Dense(128, activation='relu'),
         tf.keras.layers.Dropout(0.5),
         tf.keras.layers.Dense(64, activation='relu'),
         tf.keras.layers.Dropout(0.5),
@@ -65,4 +66,3 @@ if st.button("Send"):
             response = tf.argmax(outputs.logits, axis=-1).numpy()[0]
 
         st.write(f"Bot: {response}")
-