Update app.py

app.py

@@ -62,6 +62,9 @@ def run_classifer(model_key, image_path, prob_threshold):
     # image_path: path to test image
     # prob_threshold: confidence (i.e., probability) threshold above which to consider a prediction valid
 
+    if prob_threshold is None:
+        prob_threshold = 0.4
+    
     device = 0 if torch.cuda.is_available() else -1
 
     # Load model (cache for speed)