Soumen commited on
Commit
d309a09
1 Parent(s): 96d1891
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +0 -1
app.py CHANGED
@@ -51,4 +51,4 @@ if choice == 'Detection':
51
  preds = tokenizer.batch_decode(output_ids, skip_special_tokens=True)
52
  preds = [pred.strip() for pred in preds]
53
  return preds
54
- predict_step(our_image)
 
51
  preds = tokenizer.batch_decode(output_ids, skip_special_tokens=True)
52
  preds = [pred.strip() for pred in preds]
53
  return preds
54
+ st.success(predict_step(our_image))
requirements.txt CHANGED
@@ -1,5 +1,4 @@
1
  Pillow
2
  streamlit
3
- opencv-python
4
  transformers
5
  torch
 
1
  Pillow
2
  streamlit
 
3
  transformers
4
  torch