Update app.py

app.py

@@ -27,13 +27,17 @@ array_cols = ['array_area', *[f'mean_{i}' for i in range(1, 151)],
 expected_columns_count = len(array_cols)  # This should be 1501
 
 # Initialize PCA
-pca = PCA(n_components=10)  # Adjust n_components as needed
+pca = PCA()  # You can also specify n_components=None
 
 def get_pca_dataset(datos_df):
     """Fit PCA on the given DataFrame and return transformed data."""
-    pca.fit(datos_df)  # Fit PCA on the training data (do this once on your training dataset)
-    transformed_data = pca.transform(datos_df)  # Transform the new data
-    return transformed_data
+    # Fit PCA only on a larger dataset
+    if datos_df.shape[0] > 1:  # Only fit if we have more than one sample
+        pca.fit(datos_df)
+        transformed_data = pca.transform(datos_df)
+        return transformed_data
+    else:
+        return datos_df  # Return original data if not enough samples
 
 def process_soil_image(image):
     """Extract properties from the uploaded soil image."""