Spaces:

SubterraAI
/

ofwat_obstruction_classification_demo

Runtime error

Adonai Vera commited on Dec 4, 2023

Commit

222cac6

1 Parent(s): c6d54f2

token

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ import os
 HF_TOKEN = os.environ.get('HF_TOKEN')
 # Initialize the pipeline with your model
-pipe = pipeline("image-classification", model="SubterraAI/ofwat_cleaner_classification", token=HF_TOKEN)
 def classify_image(image):
     # Convert the input image to PIL format

 HF_TOKEN = os.environ.get('HF_TOKEN')
 # Initialize the pipeline with your model
+pipe = pipeline("image-classification", model="SubterraAI/ofwat_cleaner_classification", use_auth_token=HF_TOKEN)
 def classify_image(image):
     # Convert the input image to PIL format