Update app.py

app.py

@@ -25,7 +25,7 @@ def process_class_list(classes_string: str):
     return classes_list
 
 
-def model_inference(img, model_name: str = "yolos-small-dwr", prob_threshold: int, classes_to_show=str):
+def model_inference(img, model_name = "yolos-small-dwr", prob_threshold: int, classes_to_show=str):
     feature_extractor = AutoFeatureExtractor.from_pretrained(f"hustvl/{model_name}")
     model = YolosForObjectDetection.from_pretrained(f"hustvl/{model_name}")