Spaces:
Sleeping
Sleeping
Updated app.py file.
Browse files
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,
|