Upload folder using huggingface_hub

app.py

@@ -2,8 +2,10 @@ import gradio as gr
 from transformers import pipeline
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-# Load the Mistral-7B-v0.1 model and tokenizer
-model_name = "mistralai/Mistral-7B-v0.1"
+# Load the model and tokenizer
+# 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)