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-NeoXT-Chat-Base-20B", force_download=True, local_files_only=False)
-model = AutoModelForCausalLM.from_pretrained("togethercomputer/GPT-NeoXT-Chat-Base-20B", torch_dtype=torch.bfloat16)
+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)
 
 def gpt(prompt):
     inputs = tokenizer("<human>: Hello!\n<bot>:", return_tensors='pt').to(model.device)
@@ -15,4 +15,4 @@ def gpt(prompt):
     print(output_str)
     return output_str
 
-gr.Interface(fn=gpt,inputs="text",output="text").launch()
+gr.Interface(fn=gpt,inputs="text",outputs="text").launch()