Spaces:
Sleeping
Sleeping
File size: 470 Bytes
ef93405 1420d06 ef93405 1420d06 ef93405 1420d06 ef93405 1420d06 |
1 2 3 4 5 6 7 8 9 10 11 12 |
from transformers import pipeline
import gradio as gr
generator = pipeline("text2text-generation", model="google/flan-t5-small")
def generate_website_code(prompt):
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="text", outputs="text", title="Website Generator").launch()
|