Update app.py

app.py

@@ -1,21 +1,11 @@
-import gradio as gr
 from transformers import pipeline
+import gradio as gr
 
-# Using a lightweight model to stay within free tier limits
-generator = pipeline("text-generation", model="Salesforce/codegen2-1B", max_new_tokens=1024)
+generator = pipeline("text2text-generation", model="google/flan-t5-small")
 
 def generate_website_code(prompt):
-    system_prompt = (
-        f"Create a fully responsive website with HTML, CSS, and JavaScript. "
-        f"Use these requirements: {prompt}.\n\n"
-        f"Return full code with all parts embedded."
-    )
-    result = generator(system_prompt)[0]["generated_text"]
+    full_prompt = f"Create a stylish animated HTML, CSS, and JS website using this prompt: {prompt}"
+    result = generator(full_prompt, max_new_tokens=1024)[0]["generated_text"]
     return result
 
-gr.Interface(
-    fn=generate_website_code,
-    inputs=gr.Textbox(lines=5, label="Describe the website you want (theme, brand, images, colors)"),
-    outputs="text",
-    title="AI Website Generator"
-).launch()
+gr.Interface(fn=generate_website_code, inputs="text", outputs="text", title="Website Generator").launch()