Spaces:
Running
Running
Update Webpage.py
Browse files- Webpage.py +4 -4
Webpage.py
CHANGED
@@ -45,10 +45,10 @@ def calculate_areas(prompts, brand_num, pictorial_num, text_num):
|
|
45 |
def attention(whole_display_prompt, ad, context,
|
46 |
brand_num, pictorial_num, text_num,
|
47 |
category, ad_location, gaze_type):
|
48 |
-
text_detection_model_path = '
|
49 |
-
LDA_model_pth = '
|
50 |
-
training_ad_text_dictionary_path = '
|
51 |
-
training_lang_preposition_path = '
|
52 |
|
53 |
# caption_ad = XGBoost_utils.Caption_Generation(ad)
|
54 |
# caption_context = XGBoost_utils.Caption_Generation(context)
|
|
|
45 |
def attention(whole_display_prompt, ad, context,
|
46 |
brand_num, pictorial_num, text_num,
|
47 |
category, ad_location, gaze_type):
|
48 |
+
text_detection_model_path = 'EAST-Text-Detection/frozen_east_text_detection.pb'
|
49 |
+
LDA_model_pth = 'LDA_Model_trained/lda_model_best_tot.model'
|
50 |
+
training_ad_text_dictionary_path = 'LDA_Model_trained/object_word_dictionary'
|
51 |
+
training_lang_preposition_path = 'LDA_Model_trained/dutch_preposition'
|
52 |
|
53 |
# caption_ad = XGBoost_utils.Caption_Generation(ad)
|
54 |
# caption_context = XGBoost_utils.Caption_Generation(context)
|