More logging

app.py

@@ -6,6 +6,10 @@ import os
 import spaces
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
+from accelerate.logging import get_logger
+
+# Enable accelerate logging
+logger = get_logger(__name__, log_level="DEBUG")
 
 import huggingface_hub
 
@@ -33,14 +37,18 @@ print(f"Allocated: {torch.cuda.memory_allocated() / 1024**3:.2f} GB")
 print(f"Reserved: {torch.cuda.memory_reserved() / 1024**3:.2f} GB")
 
 try:
+    logger.info("Loading fielddecoder model...")
     fielddecoder_model = AutoModelForCausalLM.from_pretrained(
         "ejschwartz/resym-fielddecoder", 
         torch_dtype=torch.bfloat16,
         low_cpu_mem_usage=True,
         device_map="cuda",
     )
+    logger.info("Successfully loaded fielddecoder model")
 except Exception as e:
-    print(f"Error loading fielddecoder model: {e}")
+    logger.error(f"Error loading fielddecoder model: {e}")
+    import traceback
+    logger.error(traceback.format_exc())
 
 make_gradio_client = lambda: Client("https://ejschwartz-resym-field-helper.hf.space/")