Fix AttributeError: use bot.args.model instead of bot.current_model

app.py

@@ -175,6 +175,7 @@ class InferenceAPIBot:
         self.bot = bot  # Use bot for vector DB and formatting
         self.client = InferenceClient(api_key=hf_token)
         self.current_model = bot.args.model
+        self.args = bot.args  # Expose args for compatibility
         logger.info(f"InferenceAPIBot initialized with model: {self.current_model}")
     
     def generate_answer(self, prompt: str, **kwargs) -> str:
@@ -311,7 +312,11 @@ class GradioRAGInterface:
             self.bot = initial_bot
             self.use_inference_api = False
         
-        self.current_model = self.bot.current_model
+        # Get current model from bot args (not a direct attribute)
+        self.current_model = self.bot.args.model if hasattr(self.bot, 'args') else getattr(self.bot, 'current_model', None)
+        if self.current_model is None and hasattr(self.bot, 'bot'):
+            # If using InferenceAPIBot, get from the wrapped bot
+            self.current_model = self.bot.bot.args.model
         self.data_dir = initial_bot.args.data_dir
         logger.info("GradioRAGInterface initialized")
     
@@ -850,6 +855,8 @@ def create_demo_for_spaces():
         # We still need the vector database
         # Set a flag to skip model loading
         args.skip_model_loading = True
+        
+        # Create bot - it will skip model loading due to the flag
         bot = RAGBot(args)
         
         # Don't load the model - we'll use Inference API