Demo.py

import Predict

if __name__ == "__main__":
  input_ad_path = 'Demo/004.jpg'
  text_detection_model_path = 'EAST-Text-Detection/frozen_east_text_detection.pb'
  LDA_model_pth = 'LDA_Model_trained/lda_model_best_tot.model'
  training_ad_text_dictionary_path = 'LDA_Model_trained/object_word_dictionary'
  training_lang_preposition_path = 'LDA_Model_trained/dutch_preposition'

  Predict.Ad_Gaze_Prediction(input_ad_path=input_ad_path, input_ctpg_path=None, ad_location=0,
                             text_detection_model_path=text_detection_model_path, LDA_model_pth=LDA_model_pth, 
                             training_ad_text_dictionary_path=training_ad_text_dictionary_path, training_lang_preposition_path=training_lang_preposition_path, training_language='dutch', 
                             surface_sizes=None, Product_Group=None,
                             obj_detection_model_pth=None, num_topic=20, Gaze_Time_Type='Ad')