Update app.py

app.py

@@ -5,8 +5,7 @@ import gradio as gr
 import torch
 
 
-from transformers import (AutoFeatureExtractor, DetrForObjectDetection,
-                          YolosForObjectDetection)
+from transformers import (AutoFeatureExtractor, DetrForObjectDetection,)
 from visualization import visualize_attention_map, visualize_prediction
 from style import css, description, title
 
@@ -39,9 +38,6 @@ def detect_objects(model_name, image_input, threshold):
     if "DETR" in model_name:
         model = DetrForObjectDetection.from_pretrained(MODELS_REPO[model_name])
         model_details = "DETR details"
-    elif "YOLOS" in model_name:
-        model = YolosForObjectDetection.from_pretrained(MODELS_REPO[model_name])
-        model_details = "YOLOS details"
 
     (
         processed_outputs,