Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+import logging
 from typing import Dict, Any
 
 import gradio as gr
@@ -10,6 +11,7 @@ def predict(text: str) -> Dict[str, Any]:
     """Classify text for PII detection."""
     if not text or text.strip() == "":
         return {"No input provided": 0.0}
+    logging.info(f"User input: {text}")
 
     try:
         # Tokenize input
@@ -52,6 +54,12 @@ examples = [
 
 
 if __name__ == "__main__":
+    logging.basicConfig(
+        level=logging.INFO,
+        format='%(asctime)s - %(levelname)s - %(message)s',
+        handlers=[logging.StreamHandler()]
+    )
+    
     # Model configuration
     model_id = "Roblox/roblox-pii-classifier"
 
@@ -101,4 +109,4 @@ if __name__ == "__main__":
         flagging_mode="never",
     )
 
-    demo.launch()
+    demo.launch()