add torch

scripts/chatbot_logic.py

@@ -1,8 +1,9 @@
 from scripts.parsing_utils import load_yaml_file, get_roadmap_phases, get_project_rules
 import os
-from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig  # Import BitsAndBytesConfig
+from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
 import yaml
 import logging
+import torch # ADD THIS LINE - Import torch
 
 logging.basicConfig(level=logging.ERROR,
                     format='%(asctime)s - %(levelname)s - %(message)s')