Update app.py

app.py

@@ -86,6 +86,7 @@ def inference(
         target_sizes = torch.tensor(img.size[::-1]).unsqueeze(0)
         postprocessed_outputs = feature_extractor.post_process(outputs, target_sizes)
         bboxes_scaled = postprocessed_outputs[0]['boxes']
+
         
         classes_list = get_class_list_from_input(classes_to_show)
         res_img = plot_results(img, probas[keep], bboxes_scaled[keep], model, classes_list)
@@ -93,6 +94,7 @@ def inference(
         return res_img
 
         
+classes_to_show = gr.components.Textbox(placeholder="e.g. person, boat", label="Classes to use (empty means all classes)")
 
 inputs = [
     gr.inputs.Image(type="filepath", label="Input Image"),
@@ -112,9 +114,10 @@ inputs = [
     ),
     gr.inputs.Slider(minimum=0, maximum=1.0, step=0.01, default=0.9, label="Probability Threshold"),
     gr.inputs.Slider(minimum=320, maximum=1280, default=640, step=32, label="Image Size"),
-    gr.inputs.Textbox(placeholder="e.g. car, truck, traffic light", label="Classes to use (empty means all classes)"),
+    classes_to_show,
 ]
 
+
 outputs = gr.outputs.Image(type="filepath", label="Output Image")
 
 examples = [