Update app.py
Browse files
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('
|
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 |
|