Update app.py

app.py

@@ -51,7 +51,7 @@ def get_figure(in_pil_img, in_results):
     return plt.gcf()
 
 
-def infer(in_pil_img, in_model="yolos-tiny", in_threshold=0.9):
+def infer(in_pil_img, in_model="object_detection", in_threshold=0.9):
     target_sizes = torch.tensor([in_pil_img.size[::-1]])
     inputs = image_processor_tiny(images=in_pil_img, return_tensors="pt")
     outputs = model_tiny(**inputs)
@@ -78,7 +78,7 @@ with gr.Blocks(title="YOLOS Object Detection - ClassCat",
 
     gr.HTML("""<h4 style="color:navy;">1. Select a model.</h4>""")
 
-    model = gr.Radio(["yolos-tiny", "yolos-small"], value="yolos-tiny", label="Model name")
+    model = gr.Radio(['object_detection"], value="object_detection", label="Model name")
 
     gr.HTML("""<br/>""")
     gr.HTML("""<h4 style="color:navy;">2-a. Select an example by clicking a thumbnail below.</h4>""")