Test

app.py

@@ -3,7 +3,7 @@ import pandas as pd
 import numpy as np
 
 import pickle, json
-# from src.utils import *
+from src.utils import *
 
 ##### Start #####
 
@@ -34,14 +34,10 @@ examples = [
 
 def main(sentence: str, score_type: str, threshold_cosine: float, threshold_jaccard: float = 0.1):
     threshold = threshold_cosine if score_type == "cosine" else threshold_jaccard    
-    # results = search_offers(sentence, df_offers_brand_retailer, category_dict, brand_belong_category_dict, score_type, threshold)
-    results = None
+    results = search_offers(sentence, df_offers_brand_retailer, category_dict, brand_belong_category_dict, score_type, threshold)
     message, processed_results = process_output(results)
     return message, processed_results
 
-# def main(anything1, anything2, anything3, anything4):
-#     return "Just a test", None
-
 def process_output(output):
     """Function to process the output"""
     if output is None or output.empty: