Spaces:

SubterraAI
/

ofwat_obstruction_classification_demo

Runtime error

Adonai Vera commited on Dec 4, 2023

Commit

8021e2a

1 Parent(s): 222cac6

Make more test

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,8 +6,9 @@ import os
 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

 HF_TOKEN = os.environ.get('HF_TOKEN')
+print(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