Adonai Vera commited on
Commit
46a1f9a
1 Parent(s): dd904ec

Model in private repo

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="adonaivera/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")
8
 
9
  def classify_image(image):
10
  # Convert the input image to PIL format