Demo750 commited on
Commit
948bae2
·
verified ·
1 Parent(s): d85b0eb

Update Webpage.py

Browse files
Files changed (1) hide show
  1. 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 = '../XGBoost_Prediction_Model/EAST-Text-Detection/frozen_east_text_detection.pb'
49
- LDA_model_pth = '../XGBoost_Prediction_Model/LDA_Model_trained/lda_model_best_tot.model'
50
- training_ad_text_dictionary_path = '../XGBoost_Prediction_Model/LDA_Model_trained/object_word_dictionary'
51
- training_lang_preposition_path = '../XGBoost_Prediction_Model/LDA_Model_trained/dutch_preposition'
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)