Demo750 commited on
Commit
9248e53
·
verified ·
1 Parent(s): 0561bb5

Update Predict.py

Browse files
Files changed (1) hide show
  1. Predict.py +1 -1
Predict.py CHANGED
@@ -163,7 +163,7 @@ def Ad_Gaze_Prediction(input_ad_path, input_ctpg_path, ad_location,
163
  model_obj = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True, trust_repo=True, verbose=False)
164
  else:
165
  model_obj = torch.load(obj_detection_model_pth)
166
- model_lda = None #models.LdaModel.load(LDA_model_pth)
167
  dictionary = torch.load(training_ad_text_dictionary_path)
168
  dutch_preposition = torch.load(training_lang_preposition_path)
169
  ad_num_objs, ctpg_num_objs, ad_topic_weights, topic_Diff = XGBoost_utils.object_and_topic_variables(ad_img, ctpg_img, has_ctpg, dictionary,
 
163
  model_obj = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True, trust_repo=True, verbose=False)
164
  else:
165
  model_obj = torch.load(obj_detection_model_pth)
166
+ model_lda = None
167
  dictionary = torch.load(training_ad_text_dictionary_path)
168
  dutch_preposition = torch.load(training_lang_preposition_path)
169
  ad_num_objs, ctpg_num_objs, ad_topic_weights, topic_Diff = XGBoost_utils.object_and_topic_variables(ad_img, ctpg_img, has_ctpg, dictionary,