Change YOLOS input image to PIL type

Trying to see if this fixes an error with the YOLOS Image Processor

app.py

@@ -1,8 +1,9 @@
 from threading import Thread
 
+from PIL import Image
 import gradio as gr
 import torch
-from transformers import PreTrainedModel  # for type hint
+from transformers import PreTrainedModel, AutoImageProcessor  # for type hint
 from transformers import TextIteratorStreamer, AutoModelForCausalLM, AutoTokenizer  # Moondream
 from transformers import YolosImageProcessor, YolosForObjectDetection  # YOLOS-small-300
 
@@ -49,11 +50,11 @@ def answer_question(img, prompt):
         yield buffer.strip()
 
 
-def detect_objects(img):
-    inputs = yolos_processor(images=img, return_tensors="pt")
+def detect_objects(img: Image):
+    inputs = yolos_processor(img, return_tensors="pt")
     outputs = yolos_model(**inputs)
 
-    target_sizes = torch.tensor([img.size[::-1]])
+    target_sizes = torch.tensor([tuple(reversed(img.size))])
     results = yolos_processor.post_process_object_detection(outputs, threshold=0.7, target_sizes=target_sizes)[0]
 
     for score, label, box in zip(results["scores"], results["labels"], results["boxes"]):
@@ -76,7 +77,7 @@ if __name__ == "__main__":
 
         with gr.Tab("Object Detection"):
             with gr.Row():
-                yolos_input = gr.Image()
+                yolos_input = gr.Image(type="pil")
                 yolos_output = gr.Image()
             yolos_button = gr.Button("Submit")
 
@@ -89,6 +90,6 @@ if __name__ == "__main__":
                 moon_output = gr.TextArea(label="Output")
 
         moon_submit.click(answer_question, [moon_img, moon_prompt], moon_output)
-        yolos_button.click(detect_objects, [yolos_input], yolos_output)
+        yolos_button.click(detect_objects, [yolos_input])
 
     app.queue().launch()