Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import joblib
 import numpy as np
+import pandas as pd
 from huggingface_hub import hf_hub_download
 
 model = joblib.load(
@@ -15,9 +16,14 @@ def predict(*features) -> str:
     print("Features")
     print([features])
     
-    result = model.predict(
-        np.array([features])
-    )
+    # result = model.predict(
+    #     np.array([features])
+    # )
+
+    columns = ['pns','seraph','opus','draconic','null_foe','off_ele']
+    data = pd.DataFrame([features], columns=columns)
+    result = model.predict(data)
+    
     print("Result")
     print(result)