Update app.py

app.py

@@ -13,8 +13,13 @@ hf_token = os.getenv("DOWNLOAD")
 
 # Load tokenizer and model with proper token
 # Correct code to load the model and tokenizer with proper authentication and remote code trust
-tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=hf_token)
-model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=hf_token, trust_remote_code=True)
+# Install the required dependencies first
+# pip install flash_attn causal_conv1d mamba_ssm einops
+
+# Loading model and tokenizer
+tokenizer = AutoTokenizer.from_pretrained(model_name, token=hf_token)
+model = AutoModelForCausalLM.from_pretrained(model_name, token=hf_token, trust_remote_code=True)
+
 
 
 # Define your neural networks