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)