Update app.py

app.py

@@ -40,7 +40,7 @@ if torch.cuda.is_available():
     base_model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto", quantization_config=bnb_config)
     storytell_model = PeftModel.from_pretrained(base_model, "ranamhamoud/storytell")
     storytell_tokenizer = AutoTokenizer.from_pretrained(model_id)
-    storytell_tokenizer.pad_token = tokenizer.eos_token
+    storytell_tokenizer.pad_token = storytell_tokenizer.eos_token
     
     
     editing_model_id = "meta-llama/Llama-2-7b-chat-hf"