Update app.py

app.py

@@ -3,8 +3,9 @@ from transformers import pipeline
 from PIL import Image
 import requests
 from io import BytesIO
+import os
 
-
+hf_token = os.environ.get("HF_TOKEN")
 # Title
 st.title("Image Classification Web App")
 st.markdown("This app uses Hugging Face's 'transformers' library to classify images using pre-trained models. The app uses three different models for image classification: swin, convnext and vit. Please select a model to classify the image you put on the left sidebar.")
@@ -37,13 +38,13 @@ st.sidebar.markdown("*Swin Transformer* https://huggingface.co/docs/transformers
 # Image classification function
 
 def classify_image1(image):
-    pipe1 = pipeline("image-classification", "SolubleFish/swin_transformer-finetuned-eurosat")
+    pipe1 = pipeline("image-classification", "SolubleFish/swin_transformer-finetuned-eurosat", token=hf_token)
     return pipe1(image)
 def classify_image2(image):
-    pipe2 = pipeline("image-classification", "SolubleFish/image_classification_convnext")
+    pipe2 = pipeline("image-classification", "SolubleFish/image_classification_convnext", token=hf_token)
     return pipe2(image)
 def classify_image3(image):
-    pipe3 = pipeline("image-classification", "SolubleFish/image_classification_vit")
+    pipe3 = pipeline("image-classification", "SolubleFish/image_classification_vit", token=hf_token)
     return pipe3(image)