Update app.py

app.py

@@ -1,19 +1,19 @@
 import gradio as gr
 from transformers import pipeline
 
-# Using a smaller, more stable model
+# Stable configuration for CPU
 model = pipeline(
     "text-generation",
-    model="bigcode/tiny_starcoder_py",  # Very small model for CPU
+    model="bigcode/tiny_starcoder_py",
     device="cpu"
 )
 
 def generate_code(prompt):
     try:
         response = model(
-            f"# Python Question\n{prompt}\n# Answer\n",
+            f"# Question\n{prompt}\n# Answer\n",
             max_new_tokens=80,
-            temperature=0.2,
+            temperature=0.3,
             do_sample=False
         )
         return response[0]['generated_text'].split("# Answer\n")[-1]
@@ -24,8 +24,7 @@ demo = gr.Interface(
     fn=generate_code,
     inputs=gr.Textbox(lines=2, placeholder="Ask about Python..."),
     outputs=gr.Code(language="python"),
-    title="Stable Coding Assistant",
-    allow_flagging="never"
+    title="Stable Coding Assistant"
 )
 
 demo.launch()