changes made to chatbot and db

chatbot.py

@@ -3,6 +3,9 @@ import streamlit as st
 import google.generativeai as genai
 from transformers import pipeline, AutoModelForSequenceClassification, AutoTokenizer
 
+# Disable HF transfer mechanism to avoid rate limiting errors
+os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "0"
+
 # 🔑 Fetch API key from Hugging Face Secrets
 GEMINI_API_KEY = os.getenv("gemini_api")
 

db.py

@@ -3,13 +3,13 @@ import requests
 import io
 from pymongo import MongoClient
 
-#### **1. MongoDB Connection**
+# 1. MongoDB Connection
 def get_mongo_client():
     client = MongoClient("mongodb+srv://groupA:[email protected]/?retryWrites=true&w=majority&appName=SentimentCluster")
     db = client["sentiment_db"]
     return db["tweets"]
 
-#### **2. Insert Data If Collection is Empty**
+# 2. Insert Data If Collection is Empty
 def insert_data_if_empty():
     collection = get_mongo_client()