Update app.py

app.py

@@ -5,7 +5,7 @@ import os
 import torch
 import torch.nn as nn
 from transformers.activations import get_activation
-from transformers import AutoTokenizer, AutoModelWithLMHead, AutoModelForCausalLM
+from transformers import AutoTokenizer, AutoModelWithLMHead, AutoModelForCausalLM, GPTJForCausalLM
 from transformers import XGLMTokenizer, XGLMForCausalLM
 st.title('GPT2: To see all prompt outlines: https://huggingface.co/BigSalmon/InformalToFormalLincoln46')
 #device = torch.device("cuda" if torch.cuda.is_available() else "cpu")