Update app.py

app.py

@@ -4,6 +4,28 @@ Lightweight Aphasia Classification App
 Optimized for Hugging Face Spaces with lazy loading and fallbacks
 """
 
+import os
+
+# Configure environment for CPU-only and memory optimization
+os.environ['CUDA_VISIBLE_DEVICES'] = ''  # Force CPU-only
+os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'max_split_size_mb:128'
+os.environ['OMP_NUM_THREADS'] = '2'  # Limit CPU threads
+os.environ['MKL_NUM_THREADS'] = '2'
+os.environ['NUMEXPR_NUM_THREADS'] = '2'
+os.environ['TOKENIZERS_PARALLELISM'] = 'false'  # Avoid tokenizer warnings
+
+# Batchalign specific settings
+os.environ['BATCHALIGN_CACHE'] = '/tmp/batchalign_cache'
+os.environ['HF_HUB_CACHE'] = '/tmp/hf_cache'  # Use tmp for model cache
+os.environ['TRANSFORMERS_CACHE'] = '/tmp/transformers_cache'
+
+# Whisper settings for CPU optimization
+os.environ['WHISPER_CACHE'] = '/tmp/whisper_cache'
+
+print("🔧 Environment configured for CPU-only processing")
+print("💾 Model caches set to /tmp/ to save space")
+
+
 from flask import Flask, request, render_template_string, jsonify
 import os
 import tempfile