Update app_local.py

app_local.py

@@ -11,10 +11,17 @@ import re
 import gc
 import json  # Added json import
 from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
-
+import logging
 #############################
 os.environ.setdefault('GRADIO_ANALYTICS_ENABLED', 'False')
 os.environ.setdefault('HF_HUB_DISABLE_TELEMETRY', '1')
+# Set up logging
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(levelname)s - %(message)s',
+    handlers=[logging.StreamHandler(sys.stdout)]
+)
+logger = logging.getLogger(__name__)
 
 # Model configuration
 REWRITER_MODEL = "Qwen/Qwen1.5-4B-Chat"  # Upgraded to 4B for better JSON handling