vishnun commited on
Commit
82e3868
1 Parent(s): fd9c32f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -8,8 +8,9 @@ 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
 
14
  st.success('Model loaded')
15
 
 
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
+ model_id = "vishnun/codenlbert-tiny"
12
+ tokenizer = AutoTokenizer.from_pretrained(model_id)
13
+ model = AutoModelForSequenceClassification.from_pretrained(model_id)
14
 
15
  st.success('Model loaded')
16