Update app.py

app.py

@@ -2,18 +2,16 @@ import gradio as gr
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-# The public model ID
 model_id = "lakshraina2/leetcodeAI" 
 
-print("Loading model on CPU (Public Access)...")
+print("Loading model on CPU...")
 
-# Force token=False to bypass the 401 error on public repos
 tokenizer = AutoTokenizer.from_pretrained(model_id, token=False)
+# We removed device_map to avoid the Accelerate dependency error
 model = AutoModelForCausalLM.from_pretrained(
     model_id, 
-    torch_dtype=torch.float32, 
-    device_map={"": "cpu"},
-    token=False # This is the magic fix
+    dtype=torch.float32, 
+    token=False
 )
 
 def solve(problem_text):