aamirtaymoor commited on
Commit
c2cac49
·
verified ·
1 Parent(s): 570c1b6

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -43,7 +43,7 @@ def predict_all_reviews(reviews: AllTextRatingRequest):
43
  processed_data_list = list()
44
  start_time = time.time()
45
  for review in reviews:
46
- raw_data = {"text":review.text, "star_rating":review.rating, "skip":skip}
47
  processed_data, has_sentiments = process_single_comment(raw_data)
48
  porcessed_data_list.append({"processed_data":processed_data, "has_sentiments":has_sentiments})
49
  end_time = time.time()
 
43
  processed_data_list = list()
44
  start_time = time.time()
45
  for review in reviews:
46
+ raw_data = {"text":review.get('text', str()), "star_rating":review.get('rating', 5), "skip":skip}
47
  processed_data, has_sentiments = process_single_comment(raw_data)
48
  porcessed_data_list.append({"processed_data":processed_data, "has_sentiments":has_sentiments})
49
  end_time = time.time()