Spaces:

SubterraAI
/

ofwat_obstruction_classification_demo

Runtime error

Adonai Vera commited on Jan 17, 2024

Commit

06d5653

1 Parent(s): f54d064

tempo

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,8 @@ import os
 # Initialize the pipeline with your model
 pipe = pipeline("image-classification", model="SubterraAI/ofwat_cleaner_classification")
 HF_TOKEN = os.environ.get('HF_TOKEN')
-hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "ofwat_cleaner_loop", private=True, separate_dirs=True)
 def classify_image(image):
     # Convert the input image to PIL format

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