acecalisto3 commited on
Commit
e83a3b2
·
verified ·
1 Parent(s): 0a9b7ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -13
app.py CHANGED
@@ -47,19 +47,14 @@ class SchoolChatbot:
47
  self.setup_models()
48
  self.resources = []
49
  self.visited_urls = set()
50
- self.crawl_and_index_resources()
51
-
52
- def setup_models(self):
53
- """Initialize all required models"""
54
- try:
55
- logger.info("Setting up models...")
56
- self.tokenizer = AutoTokenizer.from_pretrained(Config.MODEL_NAME)
57
- self.model = AutoModelForCausalLM.from_pretrained(Config.MODEL_NAME)
58
- self.embedding_model = SentenceTransformer(Config.EMBEDDING_MODEL)
59
- logger.info("Models setup completed successfully")
60
- except Exception as e:
61
- logger.error(f"Error setting up models: {e}")
62
- raise
63
 
64
  def crawl_url(self, url: str, depth: int):
65
  """Recursively crawl URLs up to specified depth"""
 
47
  self.setup_models()
48
  self.resources = []
49
  self.visited_urls = set()
50
+ self.crawl_and_index_resources() # This line will now work
51
+
52
+ def crawl_and_index_resources(self):
53
+ """Crawl and index resources from the base URLs."""
54
+ logger.info("Starting to crawl and index resources...")
55
+ for base_url in Config.BASE_URLS:
56
+ self.crawl_url(base_url, depth=0)
57
+ logger.info("Crawling and indexing completed.")
 
 
 
 
 
58
 
59
  def crawl_url(self, url: str, depth: int):
60
  """Recursively crawl URLs up to specified depth"""