dipasha's picture
Rename collision to app.py
7d5c9e6
raw
history blame
No virus
245 Bytes
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()