vishnun commited on
Commit
fd9c32f
1 Parent(s): f83f807

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -5,8 +5,9 @@ from PIL import Image
5
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
6
 
7
  st.title(':blue[_SnapCode_]')
 
8
 
9
- with st.spinner('Model is loading'):
10
  tokenizer = AutoTokenizer.from_pretrained("vishnun/codenlbert-sm")
11
  model = AutoModelForSequenceClassification.from_pretrained("vishnun/codenlbert-sm")
12
 
 
5
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
6
 
7
  st.title(':blue[_SnapCode_]')
8
+ st.markdown("_Extract code blocks out of Screenshots and Images_")
9
 
10
+ with st.spinner('Code vs Natuaral language - Classification model is loading'):
11
  tokenizer = AutoTokenizer.from_pretrained("vishnun/codenlbert-sm")
12
  model = AutoModelForSequenceClassification.from_pretrained("vishnun/codenlbert-sm")
13