Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
|
@@ -22,8 +22,6 @@ os.makedirs(cache_dir, exist_ok=True)
|
|
| 22 |
gliner_model = GLiNER.from_pretrained("urchade/gliner_medium-v2.1",cache_dir=cache_dir)
|
| 23 |
groq_client = Groq(api_key=GROQ_API_KEY)
|
| 24 |
|
| 25 |
-
init_qdrant_collection()
|
| 26 |
-
populate_vectordb_from_hf()
|
| 27 |
def extract_entities(text):
|
| 28 |
# Tokenize the input text first
|
| 29 |
labels = ["PRODUCT", "ISSUE", "PROBLEM", "SERVICE"]
|
|
|
|
| 22 |
gliner_model = GLiNER.from_pretrained("urchade/gliner_medium-v2.1",cache_dir=cache_dir)
|
| 23 |
groq_client = Groq(api_key=GROQ_API_KEY)
|
| 24 |
|
|
|
|
|
|
|
| 25 |
def extract_entities(text):
|
| 26 |
# Tokenize the input text first
|
| 27 |
labels = ["PRODUCT", "ISSUE", "PROBLEM", "SERVICE"]
|