Tim13ekd's picture
Create app.py
eb70dfd verified
raw
history blame
940 Bytes
import gradio as gr
from transformers import pipeline
# Lade dein Modell
generator = pipeline(
"text-generation",
model="[dein-username]/HTML5-Willy-small1"
)
def generate_html(prompt):
"""Generiere HTML5 Code"""
result = generator(
prompt,
max_length=300,
temperature=0.7,
top_p=0.9,
do_sample=True
)
return result[0]['generated_text']
# Erstelle Interface
demo = gr.Interface(
fn=generate_html,
inputs=gr.Textbox(
label="HTML5 Prompt",
placeholder="z.B. <!DOCTYPE html>",
lines=3
),
outputs=gr.Textbox(
label="Generated Code",
lines=10
),
title="🎮 HTML5-Willy-small1",
description="HTML5 Code-Generator powered by AI",
examples=[
["<!DOCTYPE html>"],
["<canvas id='game'"],
["<form>"],
["body { background:"],
]
)
if __name__ == "__main__":
demo.launch()