Spaces:
Sleeping
Sleeping
Update Predict.py
Browse files- 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
|
| 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,
|