Update app.py

app.py

@@ -6,7 +6,7 @@ from transformers.generation import GenerationConfig
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 
 # Initialize model and tokenizer
-model_name_or_path = "TheBloke/OpenHermes-2-Mistral-7B-GPTQ"
+model_name_or_path = "teknium/OpenHermes-2-Mistral-7B"
 
 model = AutoModelForCausalLM.from_pretrained(model_name_or_path,
                                              device_map="auto",
@@ -15,7 +15,7 @@ model = AutoModelForCausalLM.from_pretrained(model_name_or_path,
 
 tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, use_fast=True)
 
-model.generation_config = GenerationConfig.from_pretrained("TheBloke/OpenHermes-2-Mistral-7B-GPTQ", trust_remote_code=False)
+model.generation_config = GenerationConfig.from_pretrained("teknium/OpenHermes-2-Mistral-7B", trust_remote_code=False)
 
 # Postprocess function
 def postprocess(self, y):