Update app.py

app.py

@@ -3,10 +3,10 @@ from transformers import pipeline
 from PIL import Image
 import requests
 from io import BytesIO
-import os
 
-hf_token = os.environ.get("HF_TOKEN")
-print(hf_token)
+
+hf_token = st.secrets["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.")