Create app.py
Browse files
app.py
ADDED
@@ -0,0 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import gradio as gr
|
2 |
+
from transformers import pipeline
|
3 |
+
|
4 |
+
# Carregar modelo de gera莽茫o de c贸digo
|
5 |
+
generator = pipeline("text-generation", model="bigcode/starcoder")
|
6 |
+
|
7 |
+
def gerar_site(prompt):
|
8 |
+
instrucao = f"Crie um c贸digo HTML com CSS embutido para: {prompt}"
|
9 |
+
resultado = generator(instrucao, max_length=600, temperature=0.7, do_sample=True)
|
10 |
+
codigo = resultado[0]["generated_text"]
|
11 |
+
return f"<iframe srcdoc='{codigo}' style='width:100%; height:500px;'></iframe>", codigo
|
12 |
+
|
13 |
+
# Interface
|
14 |
+
app = gr.Interface(
|
15 |
+
fn=gerar_site,
|
16 |
+
inputs=gr.Textbox(lines=3, placeholder="Ex: Site de portf贸lio com foto e bot茫o de contato"),
|
17 |
+
outputs=[gr.HTML(), gr.Code(language="html")]
|
18 |
+
)
|
19 |
+
|
20 |
+
app.launch()
|