Spaces:

hidevs-community
/

Waiter-Tips-Prediction

Sleeping

hidevscommunity commited on Jan 13, 2024

Commit

f74fecd

verified ·

1 Parent(s): d08aea5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ def load_model():
 # Function for model prediction
 def model_prediction(model, features):
-    predicted = str(model.predict(features)[0])
     return predicted
 def transform(text):

 # Function for model prediction
 def model_prediction(model, features):
+    predicted = str(model.predict(features))
     return predicted
 def transform(text):