Update app.py

app.py

@@ -25,14 +25,13 @@ class ConversationManager:
         if not model_name:
             print("Error: Empty model name provided")
             return None
-    
+
         if model_name in self.models:
             return self.models[model_name]
 
         try:
             print(f"Attempting to load model: {model_name}")
             tokenizer = AutoTokenizer.from_pretrained(model_name)
-        finally:
             # Try to load the model with 8-bit quantization
             model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto", load_in_8bit=True)
         except RuntimeError as e:
@@ -41,7 +40,12 @@ class ConversationManager:
                 model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")
             else:
                 model = AutoModelForCausalLM.from_pretrained(model_name)
-        
+        except Exception as e:
+            print(f"Failed to load model {model_name}: {e}")
+            print(f"Error type: {type(e).__name__}")
+            print(f"Error details: {str(e)}")
+            return None
+
         self.models[model_name] = (model, tokenizer)
         print(f"Successfully loaded model: {model_name}")
         return self.models[model_name]