Test with token

app.py

@@ -2,9 +2,12 @@ from turtle import title
 import gradio as gr
 from transformers import pipeline
 from PIL import Image
+import os
+
+HF_TOKEN = os.environ.get('HF_TOKEN')
 
 # Initialize the pipeline with your model
-pipe = pipeline("image-classification", model="SubterraAI/ofwat_cleaner_classification")
+pipe = pipeline("image-classification", model="SubterraAI/ofwat_cleaner_classification", token=HF_TOKEN)
 
 def classify_image(image):
     # Convert the input image to PIL format