Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| from transformers import pipeline | |
| try: | |
| generator = pipeline("text-generation", model="HTML5-Willy-small1") | |
| except: | |
| generator = pipeline("text-generation", model="gpt2") | |
| def generate_html(prompt): | |
| if not prompt.strip(): | |
| return "Bitte geben Sie einen Prompt ein." | |
| result = generator(prompt, max_length=300, num_return_sequences=1) | |
| return result[0]['generated_text'] | |
| gr.Interface( | |
| fn=generate_html, | |
| inputs=gr.Textbox(label="HTML5 Prompt", placeholder="z.B. <html>"), | |
| outputs=gr.Textbox(label="Generated HTML5 Code"), | |
| title="HTML5-Willy-small1", | |
| description="Generiert HTML5-Code basierend auf Ihrem Prompt" | |
| ).launch() | |