Spaces:

Ankush05
/

Code6

Build error

Ankush05 commited on Sep 21, 2023

Commit

a0f6eac

1 Parent(s): c7dde3b

this

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ from transformers import pipeline
 # xyz = {"This is new code678"}
 # col.insert_one(xyz)
-token_classifier = pipeline(model="Jean-Baptiste/camembert-ner", aggregation_strategy="simple")
 if ner := st.chat_input("query "):
     ans = token_classifier(ner)
     with st.chat_message("User"):

 # xyz = {"This is new code678"}
 # col.insert_one(xyz)
+token_classifier = pipeline(model="dslim/bert-base-NER", aggregation_strategy="simple")
 if ner := st.chat_input("query "):
     ans = token_classifier(ner)
     with st.chat_message("User"):