Update app.py

app.py

@@ -46,7 +46,7 @@ model.to(device)
 def sigmoid(_outputs):
     return 1.0 / (1.0 + np.exp(-_outputs))
 
-def predict(image, threshold):
+def predict(image, threshold=0.5):
     # Preprocess the image
     processor = AutoImageProcessor.from_pretrained(checkpoint_name)
     inputs = processor(images=image, return_tensors="pt").to(device)