Fix classification issue

app.py

@@ -14,7 +14,7 @@ model = AutoModelForImageClassification.from_pretrained(
     label2id={'Crack': 0, 'Red-Dots': 1, 'Toothmark': 2}
 )
 
-def classify_image(image):
+def classify_image(image, threshold=0.5):
     inputs = processor(images=image, return_tensors="pt")
     with torch.no_grad():
         outputs = model(**inputs)