Spaces:

Ayesha931
/

PM-LLMAssistant

Sleeping

Ayesha931 commited on May 14

Commit

67a7348

verified ·

1 Parent(s): 0195723

Update src/streamlit_app.py

Files changed (1) hide show

src/streamlit_app.py CHANGED Viewed

@@ -53,7 +53,7 @@ JSON_TEMPLATE = {
 # --- Load Model ---
 tokenizer = AutoTokenizer.from_pretrained("sentence-transformers/all-MiniLM-L6-v2")
-model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
 def mean_pooling(model_output, attention_mask):
     token_embeddings = model_output[0]

 # --- Load Model ---
 tokenizer = AutoTokenizer.from_pretrained("sentence-transformers/all-MiniLM-L6-v2")
+model = SentenceTransformer('sentence-transformers/paraphrase-mpnet-base-v2')
 def mean_pooling(model_output, attention_mask):
     token_embeddings = model_output[0]