deeksonparlma commited on
Commit
0fd739c
·
1 Parent(s): 4dcb30c
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -35,8 +35,8 @@ def classify_text(inp):
35
  # reshape the input to 2D
36
  # convert the input to a numpy array
37
  # return model.predict( np.array(inp).reshape(1, -1) )
38
- return model.predict([inp])
39
-
40
 
41
 
42
  # # encode the input text
 
35
  # reshape the input to 2D
36
  # convert the input to a numpy array
37
  # return model.predict( np.array(inp).reshape(1, -1) )
38
+ reshaped = np.array(inp).reshape(1, -1)
39
+ return model.predict(reshaped)
40
 
41
 
42
  # # encode the input text