dipasha's picture
Update app.py
055ed56
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()