Update main.py

main.py

@@ -7,6 +7,7 @@ from ml_service import process_single_comment
 from typing import List, Dict, Union
 import time
 from kafka import KafkaConsumer, KafkaProducer
+import asyncio
 
 
 app = FastAPI()