curtpond commited on
Commit
d75e27c
1 Parent(s): 733f064

Updated app.py file.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ model = pickle.load(open('lg_classifier.sav', 'rb'))
8
 
9
  # Define function to predict
10
  def predict(text):
11
- return model.predict([text])[0]
12
 
13
  # Define interface
14
  demo = gr.Interface(fn=predict,
 
8
 
9
  # Define function to predict
10
  def predict(text):
11
+ return model.predict([[text]])[0]
12
 
13
  # Define interface
14
  demo = gr.Interface(fn=predict,