Update app.py

app.py

@@ -10,11 +10,13 @@ df = pd.read_csv('dataset_actualizado.csv')
 
 # Escalar datos
 scaler = StandardScaler()
-X = scaler.fit_transform(df)
+X = df[['frecuencia_compra', 'monto_promedio', 'edad', 'ingreso']]
+scaler = StandardScaler()
+X_scaled = scaler.fit_transform(X)
 
 # Entrenar modelo KMeans
 kmeans = KMeans(n_clusters=3, random_state=42)
-kmeans.fit(X)
+kmeans.fit(X_scaled)
 
 # Función para predecir segmento
 def predecir_segmento(frecuencia_compra, monto_promedio, edad, ingreso):
@@ -64,3 +66,4 @@ iface = gr.Interface(
 if __name__ == "__main__":
     iface.launch()
 
+