import gradio as gr
from transformers import pipeline
collision = pipeline("image-classification")
def get_collision(input_img):
  return get_collision(input_img)