Update app.py

app.py

@@ -8,31 +8,13 @@ model = joblib.load(
 	hf_hub_download("Skreeauk/dark-gbf-xgboost2", "model.joblib")
 )
 
-print("Model")
-print(model)
-
 def predict(*features) -> str:
-
-    print("Features")
-    print([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)
+    result = model.predict(data).tolist()[0]
     
-    print("Result")
-    print(result)
-
-    return "Test"
-    # return model.predict(
-    #     np.array([features])
-    # ).tolist()[0]
-
-items = ['Seraph', 'Opus', 'Draconic', 'Null Ele Foe', 'Off Ele Foe']
+    return result
 
 demo = gr.Interface(
     fn=predict,