Update app.py

app.py

@@ -5,7 +5,11 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 model_name = "https://huggingface.co/spaces/DR-Rakshitha/wizardlm_api/blob/main/pytorch_model-00001-of-00002.bin"
 
 # Initialize the tokenizer
-tokenizer = AutoTokenizer.from_pretrained(model_name, local_files_only=True)
+# tokenizer = AutoTokenizer.from_pretrained(model_name, local_files_only=True)
+tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
+tokenizer.pad_token = tokenizer.eos_token
+tokenizer.padding_side = "right"
+
 
 # Initialize the GPT4All model
 model = AutoModelForCausalLM.from_pretrained(model_name)