Update app.py
Browse files
app.py
CHANGED
@@ -3,10 +3,10 @@ from transformers import pipeline
|
|
3 |
from PIL import Image
|
4 |
import requests
|
5 |
from io import BytesIO
|
6 |
-
import os
|
7 |
|
8 |
-
|
9 |
-
|
|
|
10 |
# Title
|
11 |
st.title("Image Classification Web App")
|
12 |
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.")
|
|
|
3 |
from PIL import Image
|
4 |
import requests
|
5 |
from io import BytesIO
|
|
|
6 |
|
7 |
+
|
8 |
+
hf_token = st.secrets["HF_TOKEN"]
|
9 |
+
|
10 |
# Title
|
11 |
st.title("Image Classification Web App")
|
12 |
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.")
|