j-higgins commited on
Commit
cacad38
·
unverified ·
1 Parent(s): 34b654b

Update app.py

Browse files

updated absolute file path for google_categories(v2).txt

Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -19,6 +19,11 @@ from gliner_spacy.pipeline import GlinerSpacy
19
  import warnings
20
  import threading
21
  warnings.filterwarnings("ignore", message="The sentencepiece tokenizer")
 
 
 
 
 
22
 
23
  # Configuration for GLiNER integration
24
  custom_spacy_config = {
@@ -67,7 +72,7 @@ def extract_entities(text):
67
  return entities if entities else ["No specific entities found"]
68
 
69
  # Load Google's content categories
70
- with open('google_categories(v2).txt', 'r') as f:
71
  google_categories = [line.strip() for line in f]
72
 
73
  # Function to precompute category embeddings
 
19
  import warnings
20
  import threading
21
  warnings.filterwarnings("ignore", message="The sentencepiece tokenizer")
22
+ import os
23
+
24
+ # At the top of your script, after imports
25
+ BASE_DIR = os.path.dirname(os.path.abspath(__file__))
26
+ CATEGORIES_FILE = os.path.join(BASE_DIR, 'google_categories(v2).txt')
27
 
28
  # Configuration for GLiNER integration
29
  custom_spacy_config = {
 
72
  return entities if entities else ["No specific entities found"]
73
 
74
  # Load Google's content categories
75
+ with open(CATEGORIES_FILE, 'r') as f:
76
  google_categories = [line.strip() for line in f]
77
 
78
  # Function to precompute category embeddings