Update app.py

app.py

@@ -1,12 +1,14 @@
+import os
 import streamlit as st
 from transformers import TFAutoModelForImageClassification, AutoFeatureExtractor
 from PIL import Image
 import numpy as np
 
 # Load the model and feature extractor
-model = TFAutoModelForImageClassification.from_pretrained("willco-afk/tree-test")  # Replace with your repo
+model = TFAutoModelForImageClassification.from_pretrained(os.environ.get("MODEL_ID", "willco-afk/tree-test-x")) 
 feature_extractor = AutoFeatureExtractor.from_pretrained(model.config._name_or_path)
 
+
 # Streamlit UI
 st.title("Christmas Tree Classifier")
 st.write("Upload an image of a Christmas tree to classify it:")