Adonai Vera commited on
Commit
316cc87
·
1 Parent(s): 8021e2a

Ready to test

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -4,11 +4,9 @@ from transformers import pipeline
4
  from PIL import Image
5
  import os
6
 
7
- HF_TOKEN = os.environ.get('HF_TOKEN')
8
-
9
  print(HF_TOKEN)
10
  # Initialize the pipeline with your model
11
- pipe = pipeline("image-classification", model="SubterraAI/ofwat_cleaner_classification", token=HF_TOKEN)
12
 
13
  def classify_image(image):
14
  # Convert the input image to PIL format
 
4
  from PIL import Image
5
  import os
6
 
 
 
7
  print(HF_TOKEN)
8
  # Initialize the pipeline with your model
9
+ pipe = pipeline("image-classification", model="SubterraAI/ofwat_cleaner_classification")
10
 
11
  def classify_image(image):
12
  # Convert the input image to PIL format