Update app.py

app.py

@@ -34,7 +34,19 @@ def classify_text(text):
         logger.info(f"Classifying text: {text}")
         predictions = classifier(text)
         logger.info(f"Predictions: {predictions}")
-        return predictions
+        
+        # Process predictions to add the custom logic
+        result = []
+        for prediction in predictions:
+            if prediction['score'] > 0.95:
+                label = "Immunization"
+            else:
+                label = "None"
+            result.append({'label': label, 'score': prediction['score']})
+        
+        logger.info(f"Processed predictions: {result}")
+        return result
+
     except Exception as e:
         logger.error(f"Error classifying text: {e}")
         return {"error": str(e)}