Update app.py

app.py

@@ -4,7 +4,7 @@ from PIL import Image
 import numpy as np
 
 # Load the model and feature extractor
-model = TFAutoModelForImageClassification.from_pretrained("your-username/your-repo-name")  # Replace with your repo
+model = TFAutoModelForImageClassification.from_pretrained("willco-afk/tree-test")  # Replace with your repo
 feature_extractor = AutoFeatureExtractor.from_pretrained(model.config._name_or_path)
 
 # Streamlit UI