Update app.py

app.py

@@ -1,25 +1,13 @@
 import gradio as gr
-import requests
-import os
+from transformers import pipeline
 
-API_URL = "https://api-inference.huggingface.co/models/bigcode/starcoder"
-headers = {"Authorization": f"Bearer {os.environ['HF_TOKEN']}"}
+# Load free AI code generation model (small and mobile-friendly)
+generator = pipeline("text-generation", model="Salesforce/codegen-350M-mono")
 
 def generate_code(prompt):
-    payload = {
-        "inputs": prompt,
-        "parameters": {"max_new_tokens": 300}
-    }
-    
-    response = requests.post(API_URL, headers=headers, json=payload)
-    
-    try:
-        generated_text = response.json()[0]["generated_text"]
-    except Exception as e:
-        generated_text = f"Error: {e}\nRaw Response: {response.text}"
-    
-    return generated_text
+    response = generator(prompt, max_length=256, do_sample=True)[0]["generated_text"]
+    return response
 
-demo = gr.Interface(fn=generate_code, inputs="text", outputs="text", title="Super Coder Bot", description="Enter instructions in plain English. Get real, production-ready code.")
+demo = gr.Interface(fn=generate_code, inputs="text", outputs="text", title="Super Coder Bot", description="Enter instructions in plain English. Get real code.")
 
 demo.launch()