Update app.py

Edited the code

app.py

@@ -1,12 +1,11 @@
-MODEL_NAME = "openai/gpt-oss-20b"
-
-from transformers import AutoModelForCausalLM, AutoTokenizer
+import gradio as gr
+from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 
-model = AutoModelForCausalLM.from_pretrained("gpt2")
-tokenizer = AutoTokenizer.from_pretrained("gpt2")
+MODEL_NAME = "openai/gpt-oss-20b"
 
 # Load tokenizer & model
 #tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
+tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 model = AutoModelForCausalLM.from_pretrained(
     MODEL_NAME,
     device_map="auto",   # Automatically use GPU if available