Update app.py

app.py

@@ -5,9 +5,20 @@ import re
 from PIL import Image
 import spaces  # Add spaces import for Hugging Face Spaces
 import os
+import sys
 
 hf_token = os.getenv("API_KEY")
 
+# Set the Hugging Face logging to error only to avoid sensitive info in logs
+logging.set_verbosity_error()
+
+# If the key is found, use it to authenticate
+if hf_token:
+    HfFolder.save_token(hf_token)  # This authenticates you for this session
+else:
+    print("No HF_KEY found. Please make sure you've set up your Hugging Face API key as an environment variable.")
+
+
 # Model information
 MODEL_ID = "DeepMount00/SmolVLM-Base-ocr_base"
 OCR_INSTRUCTION = "Sei un assistente esperto di OCR, converti il testo in formato MD."