Spaces:

Tim13ekd
/

HTML5-Willy-Demo

Sleeping

Tim13ekd commited on 26 days ago

Commit

9afc8c6

verified ·

1 Parent(s): 305ca8c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,44 +1,22 @@
 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()

 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()