Update app.py

app.py

@@ -1,18 +1,14 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
-import os
 
-# Set up the Hugging Face API token
-hf_token = os.getenv("HF_TOKEN")
-
-# Define model name and local cache directory
+# Define model name and local directory
 model_name = "facebook/chameleon-7b"
-cache_dir = "./model_cache"
+local_model_dir = "./chameleon-7b"
 
-# Load the model and tokenizer with the token and cache directory
-tokenizer = AutoTokenizer.from_pretrained(model_name, token=hf_token, cache_dir=cache_dir)
-model = AutoModelForCausalLM.from_pretrained(model_name, token=hf_token, cache_dir=cache_dir)
+# Load the model and tokenizer from the local directory
+tokenizer = AutoTokenizer.from_pretrained(local_model_dir)
+model = AutoModelForCausalLM.from_pretrained(local_model_dir)
 
 # Function to generate response
 def generate_response(input_text):