Adonai Vera commited on
Commit
22c120b
1 Parent(s): db9e244

return first version

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import pipeline
4
  from PIL import Image
5
 
6
  # Initialize the pipeline with your model
7
- pipe = pipeline("image-classification", model="SubterraAI/ofwat_material_classification")
8
 
9
  def classify_image(image):
10
  # Convert the input image to PIL format
 
4
  from PIL import Image
5
 
6
  # Initialize the pipeline with your model
7
+ pipe = pipeline("image-classification", model="SubterraAI/ofwat_material_classification", token="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")
8
 
9
  def classify_image(image):
10
  # Convert the input image to PIL format