Update app.py

app.py

@@ -1,18 +1,20 @@
-# app.py
 from transformers import pipeline
+import torch
 import gradio as gr
 
-# Use a better model trained for HTML generation
-generator = pipeline("text2text-generation", model="mrm8488/t5-base-finetuned-web-code")
+# Make sure torch is imported and available
+generator = pipeline("text2text-generation", model="google/flan-t5-base")
 
-def generate_site(prompt):
-    result = generator(prompt, max_length=2048, do_sample=True)[0]["generated_text"]
-    return result
+def generate_code(prompt):
+    output = generator(prompt, max_length=2048, do_sample=True)[0]["generated_text"]
+    return output
 
-iface = gr.Interface(fn=generate_site,
-                     inputs=gr.Textbox(lines=10, placeholder="Enter your website prompt..."),
-                     outputs="text",
-                     title="AI Website Generator",
-                     description="Enter a prompt to generate a modern HTML/CSS/JS website")
+iface = gr.Interface(
+    fn=generate_code,
+    inputs=gr.Textbox(lines=5, placeholder="Enter your website prompt here..."),
+    outputs="text",
+    title="Website Code Generator",
+    description="Enter a prompt like: 'Create a modern café website using beige and brown with animations.'"
+)
 
 iface.launch()