Update app.py

app.py

@@ -14,7 +14,7 @@ with open('model.pkl', 'rb') as f:
 
 def predict(model, sentence):
 	output = model.predict([sentence])
-	result = output.argmax().item()
+	result = output.item()
 
 	categories = {
 		0: "WEAK",