Update app.py

app.py

@@ -1,10 +1,15 @@
 import gradio as gr
-from transformers import AutoModelForCausalLM, AutoTokenizer
+# from transformers import AutoModelForCausalLM, AutoTokenizer
 
 # Load the model and tokenizer
-model_name = "codegen-350M-mono"
-model = AutoModelForCausalLM.from_pretrained(model_name)
-tokenizer = AutoTokenizer.from_pretrained(model_name)
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+model_name = "Salesforce/codegen-350M-mono"
+
+# Load tokenizer and model with authentication token
+tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=True)
+model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=True)
+
 
 def explain_code(code):
     inputs = tokenizer(code, return_tensors="pt")