Update app.py

app.py

@@ -21,14 +21,14 @@ def main():
     local_model_path = Path.home().joinpath('mistral_models', model_id)
     local_model_path.mkdir(parents=True, exist_ok=True)
 
-    snapshot_download(repo_id=model_id, allow_patterns=["*.bin", "*.json", "*.model"], local_dir=local_model_path, use_auth_token=hf_token)
+    snapshot_download(repo_id=model_id, local_dir=local_model_path, use_auth_token=hf_token)
     st.success("Model downloaded successfully!")
     
     # Load the model and tokenizer
     st.text("Loading model...")
     tokenizer = AutoTokenizer.from_pretrained(local_model_path, use_auth_token=hf_token)
     model = AutoModelForCausalLM.from_pretrained(local_model_path, use_auth_token=hf_token)
-    generator = pipeline("text-generation", model=model, tokenizer=tokenizer, use_auth_token=hf_token)
+    generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
     st.success("Model loaded successfully!")
     
     user_input = st.text_area("Enter your instruction", "Explain Machine Learning to me in a nutshell.")