Update main.py

main.py

@@ -66,6 +66,7 @@ def predict_file_responses(file: UploadFile = File(...)):
     except Exception as e:
         raise HTTPException(status_code=400, detail=f"Error processing CSV file: {e}")
     processed_data_list = list()
+    start_time = time.time()
     for index, row in df.iterrows():
         try:
             text = row['ACTUAL REVIEW']
@@ -73,10 +74,11 @@ def predict_file_responses(file: UploadFile = File(...)):
             review_id = row['REVIEWID']
             raw_data = {"text":text, "star_rating":star_rating, "skip":False}
             processed_data, has_sentiments = process_single_comment(raw_data) 
-            print(f"Raw data for row {index} : {raw_data}")
             processed_data_list.append({"processed_data":processed_data, "has_sentiments":has_sentiments, "review_id":review_id})
         finally:
             pass
+    end_time = time.time()
+    print(f">>>>>>>>>>>>>>>>>>>>>>>> Processing time : {end_time - start_time} seconds >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>")
     return processed_data_list
 
 app.mount("/", StaticFiles(directory="static", html=True), name="static")