Update app.py

app.py

@@ -2,20 +2,20 @@ import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import gradio as gr
 
-# Load model
-model_id = "bigcode/starcoderbase"
+# Load DeepSeek Coder
+model_id = "deepseek-ai/deepseek-coder-1.3b-base"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id, trust_remote_code=True)
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model.to(device)
 
-# Inference function
+# Debugging function
 def debug_code(code, language):
-    prompt = f"### Debug the following {language} code:\n{code}\n### Fixed Code and Explanation:\n"
+    prompt = f"### Task: Fix and explain the following {language} code:\n{code}\n### Output:\n"
     inputs = tokenizer(prompt, return_tensors="pt").to(device)
     outputs = model.generate(**inputs, max_new_tokens=300, pad_token_id=tokenizer.eos_token_id)
-    result = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    return result[len(prompt):].strip()
+    response = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    return response[len(prompt):].strip()
 
 # Gradio UI
 gr.Interface(
@@ -25,7 +25,8 @@ gr.Interface(
         gr.Dropdown(["Python", "C", "C++", "JavaScript"], label="Code Language", value="Python")
     ],
     outputs=gr.Textbox(label="Fixed Code & Explanation"),
-    title="🛠 Eternos Debugger — Powered by StarCoder",
-    description="Paste buggy code, select language, and get a fix + explanation.",
+    title="🛠 Eternos Debugger — DeepSeek Edition",
+    description="Paste code, select language, and get a fix + explanation. Powered by DeepSeek Coder.",
     theme="default"
 ).launch()
+