dipasha commited on
Commit
539b25d
1 Parent(s): 1f49d9a

Update collision

Browse files
Files changed (1) hide show
  1. collision +3 -1
collision CHANGED
@@ -2,4 +2,6 @@ import gradio as gr
2
  from transformers import pipeline
3
  collision = pipeline("image-classification")
4
  def get_collision(input_img):
5
- return get_collision(input_img)
 
 
 
2
  from transformers import pipeline
3
  collision = pipeline("image-classification")
4
  def get_collision(input_img):
5
+ return collision(input_img)
6
+ iface = gr.Interface(fn=get_collision,inputs="image", outputs="text")
7
+ iface.launch()