Update app.py

app.py

@@ -1,13 +1,13 @@
 import gradio as gr
-from transformers import GPT2LMHeadModel, GPT2Tokenizer
+from transformers import LlamaForCausalLM, LlamaTokenizer
 import torch
 
 # Define model name and local directory
 local_model_dir = "./chameleon-7b"
 
 # Load the model and tokenizer from the local directory
-tokenizer = GPT2Tokenizer.from_pretrained(local_model_dir)
-model = GPT2LMHeadModel.from_pretrained(local_model_dir)
+tokenizer = LlamaTokenizer.from_pretrained(local_model_dir)
+model = LlamaForCausalLM.from_pretrained(local_model_dir)
 
 # Function to generate response
 def generate_response(input_text):