Update app.py

app.py

@@ -1,11 +1,5 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
-
-# Load the model and tokenizer
-model_name = "meta-llama/CodeLlama-7b-hf"
-model = AutoModelForCausalLM.from_pretrained(model_name)
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-
 import os
 
 # Check if the token is being accessed
@@ -15,6 +9,11 @@ if hf_token:
 else:
     print("Failed to retrieve Hugging Face token.")
 
+# Load the model and tokenizer
+model_name = "meta-llama/CodeLlama-7b-hf"
+model = AutoModelForCausalLM.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+
 
 def generate_code(prompt):
     inputs = tokenizer(prompt, return_tensors="pt")