fixed output

app.py

@@ -3,18 +3,10 @@ from flask import Flask, request, jsonify
 import os
 from wtforms import Form, StringField
 from wtforms.validators import DataRequired
-from config import model_ckpt, pipe, labels
+from config import model_ckpt, pipe, labels, THRESHOLD
 
 app = Flask(__name__)
 
-# # configure logging
-# logging.basicConfig(
-#     filename='app.log',
-#     level=logging.INFO,
-#     format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
-# )
-# logger = logging.getLogger(__name__)
-
 class PredictForm(Form):
     text = StringField('text', [DataRequired()])
 
@@ -29,13 +21,19 @@ def predict(text: str) -> dict:
         if preds:
             pred = preds[0]
             pred = sorted(pred, key=lambda x: x['score'], reverse=True)
-            return {labels.get(p["label"],p["label"]): float(p["score"]) for p in pred[:1]}
+            if pred[0]["score"] > THRESHOLD:
+                return {labels.get(p["label"],p["label"]): float(p["score"]) for p in pred[:1]}
+            else:
+                score = pred[0]["score"]     
+                logger.error("Prediction score below threshold. text: %s, score: %s", text, score)
+                return {'error': "Prediction score below threshold"}
         else:
             return {}
     except Exception as e:
         logger.error("Error processing request: %s", str(e))
         return {'error': str(e)}, 500
 
+
 @app.route('/language', methods=['POST'])
 def predict_language():
     """
@@ -57,11 +55,11 @@ def predict_language():
             description: Invalid request
         500:
             description: Internal server error
+        400:
+            description: Prediction score below threshold
     """
-    # form = PredictForm(request.form)
-    # if form.validate():
-    text = request.json['text']
-    if not text:
+    text = request.json.get('text')
+    if not text or len(text)==0:
         return jsonify({'error': 'Empty text provided'}), 400
 
     result = predict(text)
@@ -69,8 +67,7 @@ def predict_language():
         return jsonify(result)
     else:
         return jsonify({'error': 'No predictions found'}), 400
-    # else:
-    #     return jsonify({'error': 'Invalid input provided'}), 400
+
 
 if __name__ == '__main__':
     log_file = 'app.log'
@@ -78,4 +75,3 @@ if __name__ == '__main__':
     logger = logging.getLogger(__name__)
     logger.info("Running the app...")
     app.run()
-

config.py

@@ -1,5 +1,6 @@
 from transformers import pipeline
 
+THRESHOLD = 0.80
 model_ckpt = "papluca/xlm-roberta-base-language-detection"
 pipe = pipeline("text-classification", model=model_ckpt)