Attemp using Sentence Transformer

app.py

@@ -13,11 +13,11 @@ try:
     MODEL_NAME = "Sid-the-sloth/leetcode_unixcoder_final"
     device="cpu"
     # tokenizer=AutoTokenizer.from_pretrained(MODEL_NAME)
-    embedding_model=AutoModel.from_pretrained(MODEL_NAME)
-    pooling_model=models.Pooling(embedding_model.get_word_embedding_dimension())
+    # embedding_model=AutoModel.from_pretrained(MODEL_NAME)
+    # pooling_model=models.Pooling(embedding_model.get_word_embedding_dimension())
 
-    model=SentenceTransformer(
-        modules=[embedding_model,pooling_model],
+    model=SentenceTransformer(MODEL_NAME,
+        # modules=[embedding_model,pooling_model],
         device=device
     )