netradrishti
/

u2net-saliency

Feature Extraction

Model card Files Files and versions Community

reeteshmukul commited on Feb 28, 2023

Commit

30bae6f

·

1 Parent(s): b0e4c30

using two image files only

Files changed (1) hide show

app.py +1 -1

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ import gradio as gr
 print('Loading model...')
 model = get_u2net_model()
 print('Successfully loaded model...')
-examples = ['examples/1.jpg', 'examples/2.jpg', 'examples/3.jpg', 'examples/4.jpg','examples/5.jpg','examples/6.jpg']
 def infer(image):

 print('Loading model...')
 model = get_u2net_model()
 print('Successfully loaded model...')
+examples = ['examples/1.jpg', 'examples/6.jpg']
 def infer(image):