Spaces:

SubterraAI
/

ofwat_obstruction_classification_demo

Runtime error

Adonai Vera commited on Jan 17, 2024

Commit

41415df

1 Parent(s): a20b345

re-lunch

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,6 @@ pipe = pipeline("image-classification", model="SubterraAI/ofwat_cleaner_classifi
 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
     PIL_image = Image.fromarray(image).convert('RGB')

 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
     PIL_image = Image.fromarray(image).convert('RGB')