Spaces:

vishnun
/

SnapCode

Build error

vishnun commited on Jul 29, 2023

Commit

fc9077c

1 Parent(s): 9ef43fe

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,8 +7,8 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 st.title(':blue[_SnapCode_]')
 with st.spinner('Model is loading'):
-    tokenizer = AutoTokenizer.from_pretrained("usvsnsp/code-vs-nl")
-    model = AutoModelForSequenceClassification.from_pretrained("usvsnsp/code-vs-nl")
 st.success('Model loaded')

 st.title(':blue[_SnapCode_]')
 with st.spinner('Model is loading'):
+    tokenizer = AutoTokenizer.from_pretrained("vishnun/codenlbert")
+    model = AutoModelForSequenceClassification.from_pretrained("vishnun/codenlbert")
 st.success('Model loaded')