Update app.py

app.py

@@ -27,7 +27,7 @@ def index():
 def predict():
     
     # try:
-    print('before_hello',file=sys.stderr)
+
     input_text = request.get_json(force=True)['input']
     print(input_text,file=sys.stderr)
 
@@ -43,7 +43,7 @@ def predict():
     print(pipeline('hello'),file=sys.stderr)
     y_pred = pipeline(input_text)
     
-    return jsonify({'prediction': y_pred[0]})
+    return y_pred[0]['label']
     
     # except Exception as e:
     #     return jsonify({'error': str(e)})