Spaces:

devXpert
/

devXpert_space

Sleeping

devXpert commited on Jul 21

Commit

3df63a5

verified ·

1 Parent(s): 421a1ef

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,33 +1,22 @@
-from transformers import pipeline
 import gradio as gr
-# Load code generation model (switch to GPU if possible in settings)
-generator = pipeline("text-generation", model="deepseek-ai/deepseek-coder-6.7b-instruct")
-# Function to generate website code from user prompt
-def generate_website_code(prompt):
-    full_prompt = f"""You are a senior web developer. Based on the following instructions, generate a full HTML5 website including CSS and JavaScript inside <style> and <script> tags.
-Instructions:
-{prompt}
-Return only the complete code inside <html>...</html> tags, ready to save as an .html file.
-"""
-    output = generator(full_prompt, max_new_tokens=2048, temperature=0.7)[0]["generated_text"]
-    # Extract only the part starting with <html> to avoid model's intro text
-    html_start = output.find("<html")
-    if html_start != -1:
-        return output[html_start:]
-    return output
-# Gradio UI
-iface = gr.Interface(
-    fn=generate_website_code,
-    inputs=gr.Textbox(lines=8, label="Describe your website (Prompt)"),
-    outputs=gr.Code(label="Generated HTML Website Code"),
-    title="AI Website Generator",
-    description="Enter a prompt describing your website. Example: 'Create a modern cafe website using beige and brown colors. The brand is BrewNest. Include food items, transitions, and animations.'",
-)
-iface.launch()

 import gradio as gr
+from transformers import pipeline
+# Load the text-to-text generation model
+generator = pipeline("text2text-generation", model="google/flan-t5-small")
+# Define the function to generate website code
+def generate_website(prompt):
+    # Instruction prompt to ensure web-style code output
+    full_prompt = f"Generate a modern HTML, CSS, and JavaScript website. {prompt}"
+    result = generator(full_prompt, max_length=1024, do_sample=True)[0]['generated_text']
+    return result
+# Build a simple Gradio UI
+gr.Interface(
+    fn=generate_website,
+    inputs=gr.Textbox(label="Describe the website you want", placeholder="e.g. A modern cafe website using beige and brown colors..."),
+    outputs=gr.Textbox(label="Generated Website Code"),
+    title="AI Website Code Generator",
+    description="Enter a prompt to generate a full website (HTML/CSS/JS)."
+).launch()