Update app.py

app.py

@@ -2,13 +2,12 @@ import gradio as gr
 from huggingface_hub import hf_hub_download
 from llama_cpp import Llama
 
-# 1. Download your specific GGUF file
 model_path = hf_hub_download(
     repo_id="nihardon/fine-tuned-unit-test-generator",
-    filename="llama-3-8b.Q4_K_M.gguf",  # <--- DOUBLE CHECK THIS NAME!
+    filename="llama-3-8b.Q4_K_M.gguf", 
 )
 
-# 2. Load the Model (CPU Optimized)
+# Load the model (CPU optimized)
 llm = Llama(
     model_path=model_path,
     n_ctx=2048,
@@ -29,7 +28,7 @@ You are an expert Python QA engineer. Write a pytest unit test for the following
     output = llm(prompt, max_tokens=256, stop=["### Instruction:"], echo=False)
     return output['choices'][0]['text'].strip()
 
-# 3. The UI
+# UI
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 🧪 AI Unit Test Generator")
     gr.Markdown("**Model:** Custom Fine-Tuned Llama-3 (GGUF) | **Status:** Running Locally")