Update main.py

main.py

@@ -274,7 +274,8 @@ def find_related_products(sub_sku=None, product_name=None,forecast_data=None):
 async def get_product_count_prediction_from_DB(
                                                b_id: int,
                                                search_sku: str = Query(None, description="The SKU of the product to search"),
-                                               product_name: str = Query(None, description="The name of the product to search"):
+                                               product_name: str = Query(None, description="The name of the product to search")
+                                            ):
     
         related_products = find_related_products(sub_sku=search_sku, product_name=product_name,forecast_data = get_data(b_id,1,2,True))
         response_content = {