Update app.py

app.py

@@ -5,8 +5,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 
 os.environ["CURL_CA_BUNDLE"]=""
 
-tokenizer = AutoTokenizer.from_pretrained("togethercomputer/GPT-JT-Moderation-6B", force_download=True, local_files_only=False)
-model = AutoModelForCausalLM.from_pretrained("togethercomputer/GPT-JT-Moderation-6B", torch_dtype=torch.bfloat16)
+tokenizer = AutoTokenizer.from_pretrained("togethercomputer/GPT-JT-6B-v0", force_download=True, local_files_only=False)
+model = AutoModelForCausalLM.from_pretrained("togethercomputer/GPT-JT-6B-v0", torch_dtype=torch.bfloat16)
 
 def gpt(prompt):
     inputs = tokenizer("<human>: Hello!\n<bot>:", return_tensors='pt').to(model.device)