Update app.py

app.py

@@ -3,6 +3,14 @@ import streamlit as st
 from transformers import TFAutoModelForImageClassification, AutoFeatureExtractor
 from PIL import Image
 import numpy as np
+from huggingface_hub import login
+
+
+# Authenticate with Hugging Face token (if available)
+hf_token = os.environ.get("HF_TOKEN") 
+if hf_token:
+    login(token=hf_token)
+
 
 # Load the model and feature extractor
 model = TFAutoModelForImageClassification.from_pretrained(os.environ.get("MODEL_ID", "willco-afk/tree-test-x"))