Update app.py

app.py

@@ -4,10 +4,10 @@ from os import system
 system("pip3 install torch")
 system("pip3 install transformers")
 
-from transformers import AutoTokenizer,OpenAIGPTLMHeadModel
+from transformers import AutoTokenizer,AutoModelForCausalLM
 
-tokenizer = AutoTokenizer.from_pretrained("openai-community/openai-gpt")
-model = OpenAIGPTLMHeadModel.from_pretrained("openai-community/openai-gpt")
+tokenizer = AutoTokenizer.from_pretrained("apple/OpenELM-270M")
+model = openelm_270m = AutoModelForCausalLM.from_pretrained("apple/OpenELM-270M", trust_remote_code=True)
 
 def work(inp_text):
     out = tokenizer.encode(inp_text,return_tensors="pt")