Update app.py

app.py

@@ -2,30 +2,30 @@ import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import gradio as gr
 
-# Load a lightweight coding model
-model_id = "replit/code-v1-3b"
+# Load model
+model_id = "bigcode/starcoderbase"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
-model = AutoModelForCausalLM.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)
 
-# Debug function
+# Inference function
 def debug_code(code, language):
-    prompt = f"### Debug this {language} code:\n{code}\n### Provide fixed code and explanation:\n"
+    prompt = f"### Debug the following {language} code:\n{code}\n### Fixed Code and Explanation:\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
+    return result[len(prompt):].strip()
 
-# UI layout
+# Gradio UI
 gr.Interface(
     fn=debug_code,
     inputs=[
         gr.Textbox(lines=12, label="Paste Your Buggy Code Here"),
         gr.Dropdown(["Python", "C", "C++", "JavaScript"], label="Code Language", value="Python")
     ],
-    outputs=gr.Textbox(label="Debugged Code & Explanation"),
-    title="🛠 Eternos Basic Debugger",
-    description="Paste your code, choose language, and let AI fix & explain the bugs.",
+    outputs=gr.Textbox(label="Fixed Code & Explanation"),
+    title="🛠 Eternos Debugger — Powered by StarCoder",
+    description="Paste buggy code, select language, and get a fix + explanation.",
     theme="default"
 ).launch()