Update app.py

app.py

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