Upload folder using huggingface_hub

app.py

@@ -3,8 +3,8 @@ from transformers import pipeline
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
 # Load the model and tokenizer
-# model_name = "AdaptLLM/law-LLM"
-model_name = "google/gemma-2b"
+model_name = "AdaptLLM/law-LLM"
+# model_name = "google/gemma-2b"
 # model_name = "mistralai/Mistral-7B-v0.1"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)