Update app.py

app.py

@@ -5,7 +5,9 @@ from huggingface_hub import InferenceClient
 For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
 """
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
-
+# Load model directly
+from transformers import AutoModel
+model = AutoModel.from_pretrained("Zyphra/BlackMamba-2.8B")
 
 def respond(
     message,