updated requirements and application

app.py

@@ -55,6 +55,7 @@ def load_tokenizer_and_model():
         logger.error(f"Error loading model or tokenizer: {str(e)}")
         return None, None
 
+
 def get_embeddings(model):
     return model.get_input_embeddings().weight.data
 
@@ -186,6 +187,9 @@ def generate_definition_tree_placeholder(selected_sae, feature_number, weight_ty
 
 def gradio_interface():
     model, tokenizer = load_tokenizer_and_model()
+    if model is None or tokenizer is None:
+        return gr.Interface(lambda: "Failed to load model and tokenizer. Please check the logs for more details.", inputs=[], outputs="text")
+
     embeddings = get_embeddings(model)
 
     with gr.Blocks() as demo:

requirements.txt

@@ -11,3 +11,4 @@ matplotlib
 spaces
 graphviz
 Pillow
+accelerate