Update app.py

app.py

@@ -4,7 +4,11 @@ from transformers import AutoModelForCausalLM
 
 def LMmodel(message, history):
     base_model_id="LeoLM/leo-mistral-hessianai-7b-chat"
-    base_model = AutoModelForCausalLM.from_pretrained(base_model_id, device_map="auto",trust_remote_code=True)
+    try:
+        base_model = AutoModelForCausalLM.from_pretrained(base_model_id, device_map="auto",trust_remote_code=True)
+    except Exception as e:
+        exp = str(e)
+        return exp
     return "works"