Upload model (1).py

model (1).py

@@ -0,0 +1,15 @@
+# model.py
+import numpy as np
+from geo_metals_model_fixed import model  # Import the reconstructed model pipeline
+
+def predict_metals(input_df):
+    preds = model.predict(input_df)[0]
+    preds = np.clip(preds, 0, None)  # Ensure no negative concentrations
+    return {
+        "Fe_ppm": preds[0],
+        "Cr_ppm": preds[1],
+        "Mn_ppm": preds[2],
+        "Mo_ppm": preds[3],
+        "In_ppm": preds[4],
+        "Ta_ppm": preds[5],
+    }