Spaces:
Running
Running
Update Webpage.py
Browse files- Webpage.py +4 -4
Webpage.py
CHANGED
@@ -56,10 +56,10 @@ def calculate_areas(prompts, brand_num, pictorial_num, text_num):
|
|
56 |
def attention(notes, whole_display_prompt,
|
57 |
brand_num, pictorial_num, text_num,
|
58 |
category, ad_location, gaze_type):
|
59 |
-
text_detection_model_path = '
|
60 |
-
LDA_model_pth = '
|
61 |
-
training_ad_text_dictionary_path = '
|
62 |
-
training_lang_preposition_path = '
|
63 |
|
64 |
prod_group = np.zeros(38)
|
65 |
prod_group[GENERAL_CATEGORY[category]] = 1
|
|
|
56 |
def attention(notes, whole_display_prompt,
|
57 |
brand_num, pictorial_num, text_num,
|
58 |
category, ad_location, gaze_type):
|
59 |
+
text_detection_model_path = 'EAST-Text-Detection/frozen_east_text_detection.pb'
|
60 |
+
LDA_model_pth = 'LDA_Model_trained/lda_model_best_tot.model'
|
61 |
+
training_ad_text_dictionary_path = 'LDA_Model_trained/object_word_dictionary'
|
62 |
+
training_lang_preposition_path = 'LDA_Model_trained/dutch_preposition'
|
63 |
|
64 |
prod_group = np.zeros(38)
|
65 |
prod_group[GENERAL_CATEGORY[category]] = 1
|