Model in private repo

app.py

@@ -4,7 +4,7 @@ from transformers import pipeline
 from PIL import Image
 
 # Initialize the pipeline with your model
-pipe = pipeline("image-classification", model="adonaivera/ofwat_defects_classification")
+pipe = pipeline("image-classification", model="SubterraAI/ofwat_defects_classification")
 
 def classify_image(image):
     # Convert the input image to PIL format
@@ -28,7 +28,7 @@ iface = gr.Interface(
         ["examples/RC.jpg"]
     ],
     description="Upload an image to classify its material.",
-    title="Material Classification with AI by Subterra"
+    title="Defects Classification with AI by Subterra"
 )
 
 # Launch the interface