Commit f9c12b52 authored by Bhargava Rellu's avatar Bhargava Rellu

Optimize imports

parent 16be44d8
...@@ -7,14 +7,13 @@ from src.reviewsense_ecom.service.ReviewRater import ReviewRater ...@@ -7,14 +7,13 @@ from src.reviewsense_ecom.service.ReviewRater import ReviewRater
class ReviewService: class ReviewService:
"""Service class for handling product reviews""" """Service class for handling product reviews"""
def __init__(self, embedding_model="sentence-transformers/all-mpnet-base-v2"): def __init__(self):
""" """
Initialize the review service with MongoDB connection. Initialize the review service with MongoDB connection.
""" """
# Vector-based processing components # Vector-based processing components
self.embedding_model = embedding_model
self.rater = ReviewRater() self.rater = ReviewRater()
def _get_collection(self, collection_name): def _get_collection(self, collection_name):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment