Spaces:

codemogul
/

tabibu-mh

Sleeping

deeksonparlma commited on Feb 21, 2023

Commit

4dcb30c

1 Parent(s): c83cbb4

reshape

Files changed (1) hide show

app.py CHANGED Viewed

@@ -35,8 +35,7 @@ def classify_text(inp):
     # reshape the input to 2D
     # convert the input to a numpy array
     # return model.predict( np.array(inp).reshape(1, -1) )
-    return model.predict( np.array(inp, dtype=float).reshape(1, -1) )[0]

     # reshape the input to 2D
     # convert the input to a numpy array
     # return model.predict( np.array(inp).reshape(1, -1) )
+    return model.predict([inp])