Update modules/huggingface_inference.py

modules/huggingface_inference.py

@@ -7,11 +7,11 @@ def predict_alerts(df, hf_url, hf_token):
     for _, row in df.iterrows():
         input_payload = {
             "inputs": {
-                "solar": row["Solar_Gen__c"],
-                "wind": row["Wind_Gen__c"],
-                "tilt": row["Tilt__c"],
-                "vibration": row["Vibration__c"],
-                "camera": row["Camera_Status__c"]
+                "solar": row.get("Solar_Generation__c"),
+                "wind": row.get("Wind_Generation__c"),
+                "tilt": row.get("Tilt__c"),
+                "vibration": row.get("Vibration__c"),
+                "camera": row.get("Camera_Status__c")
             }
         }
         response = requests.post(hf_url, headers=headers, json=input_payload)