updated main.py

main.py

@@ -71,6 +71,8 @@ def startup_event():
         print(f"✅ Model '{MODEL_NAME}' and tokenizer loaded successfully.")
     except Exception as e:
         print(f"❌ Critical Error loading Hugging Face model: {e}")
+        # Make sure the model object is None if loading fails
+        model = None 
 
     try:
         client = MongoClient(MONGO_URI)
@@ -79,6 +81,7 @@ def startup_event():
         print("🚀 Successfully connected to MongoDB.")
     except Exception as e:
         print(f"❌ Critical Error connecting to MongoDB: {e}")
+        collection = None
 
 # --- API Data Models ---
 class QueryRequest(BaseModel):