Update app.py

app.py

@@ -130,7 +130,7 @@ def create_and_train_model(
 ) -> tf.keras.Model:
     """Creates and trains a neural network for classification."""
     model = Sequential([
-        Input(shape=(len(embeddings[0]),),
+        Input(shape=(len(embeddings[0]),),  # Fixed the double comma here
         Dense(64, activation='relu'),
         Dense(32, activation='relu'),
         Dense(num_classes, activation='softmax')
@@ -149,7 +149,6 @@ def create_and_train_model(
     encoded_labels = to_categorical(labels, num_classes=num_classes)
     model.fit(np.array(embeddings), encoded_labels, epochs=epochs, batch_size=batch_size, verbose=0)
     return model
-
 # --- RAG Question Answering ---
 st.header("RAG Question Answering")
 rag_model_name = st.selectbox("Select model for RAG:", ["gemini-pro"], index=0)