Hugging Face's logo

Spaces:
SubterraAI
/
ofwat_obstruction_classification_demo
Runtime error

App Files Community
Adonai Vera commited on
Commit
9cf13b2
·
1 Parent(s): 22e3686

Add secret token

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -7,7 +7,8 @@ import os
7
 
8
  # Initialize the pipeline with your model
9
  pipe = pipeline("image-classification", model="SubterraAI/ofwat_cleaner_classification")
10
- hf_writer = gr.HuggingFaceDatasetSaver(dataset_name="ofwat_cleaner_loop", private=True, separate_dirs=True)
 
11
 
12
 
13
  def classify_image(image):
 
7
 
8
  # Initialize the pipeline with your model
9
  pipe = pipeline("image-classification", model="SubterraAI/ofwat_cleaner_classification")
10
+ HF_TOKEN = os.environ.get('HF_TOKEN')
11
+ hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, dataset_name="ofwat_cleaner_loop", private=True, separate_dirs=True)
12
 
13
 
14
  def classify_image(image):