Update app.py

app.py

@@ -8,8 +8,9 @@ st.title(':blue[_SnapCode_]')
 st.markdown("_Extract code blocks out of Screenshots and Images_")
 
 with st.spinner('Code vs Natuaral language - Classification model is loading'):
-    tokenizer = AutoTokenizer.from_pretrained("vishnun/codenlbert-sm")
-    model = AutoModelForSequenceClassification.from_pretrained("vishnun/codenlbert-sm")
+    model_id = "vishnun/codenlbert-tiny"
+    tokenizer = AutoTokenizer.from_pretrained(model_id)
+    model = AutoModelForSequenceClassification.from_pretrained(model_id)
 
 st.success('Model loaded')