File size: 245 Bytes
1f49d9a
 
 
 
bb8a646
539b25d
 
1
2
3
4
5
6
7
import gradio as gr
from transformers import pipeline
collision = pipeline("image-classification")
def get_collision(input_img):
    return collision(input_img)
iface = gr.Interface(fn=get_collision,inputs="image", outputs="text")
iface.launch()