Update app.py

Import AutoModelForCasualLM

app.py

@@ -1,5 +1,10 @@
 MODEL_NAME = "openai/gpt-oss-20b"
 
+from transformers import AutoModelForCausalLM, AutoTokenizer
+
+model = AutoModelForCausalLM.from_pretrained("gpt2")
+tokenizer = AutoTokenizer.from_pretrained("gpt2")
+
 # Load tokenizer & model
 #tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 model = AutoModelForCausalLM.from_pretrained(