Update app.py

app.py

@@ -41,7 +41,7 @@ def get_model():
     #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln61Paraphrase")
     #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln66Paraphrase")
     tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln65Paraphrase")
-    model = AutoModelForCausalLM.from_pretrained("BigSalmon/SmallerGPT2InformalToFormalLincoln67Paraphrase")
+    model = GPTJForCausalLM.from_pretrained("EleutherAI/gpt-j-6B", revision="float16", torch_dtype=torch.float16, low_cpu_mem_usage=True)
     return model, tokenizer
     
 model, tokenizer = get_model()