Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -7,18 +7,12 @@ model = AutoModelForSeq2SeqLM.from_pretrained("t5-small")
|
|
7 |
|
8 |
# Função para gerar SQL a partir de linguagem natural
|
9 |
def gerar_sql(consulta_natural):
|
10 |
-
# Tokenizar a consulta em linguagem natural
|
11 |
inputs = tokenizer(consulta_natural, return_tensors="pt", max_length=512, truncation=True)
|
12 |
-
|
13 |
-
# Gerar a consulta SQL
|
14 |
outputs = model.generate(**inputs, max_length=512)
|
15 |
-
|
16 |
-
# Decodificar a saída para texto
|
17 |
sql_gerado = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
18 |
-
|
19 |
return sql_gerado
|
20 |
|
21 |
-
# Interface Gradio
|
22 |
interface = gr.Interface(
|
23 |
fn=gerar_sql,
|
24 |
inputs="text",
|
@@ -28,4 +22,4 @@ interface = gr.Interface(
|
|
28 |
)
|
29 |
|
30 |
# Iniciar a interface
|
31 |
-
interface.launch()
|
|
|
7 |
|
8 |
# Função para gerar SQL a partir de linguagem natural
|
9 |
def gerar_sql(consulta_natural):
|
|
|
10 |
inputs = tokenizer(consulta_natural, return_tensors="pt", max_length=512, truncation=True)
|
|
|
|
|
11 |
outputs = model.generate(**inputs, max_length=512)
|
|
|
|
|
12 |
sql_gerado = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
|
|
13 |
return sql_gerado
|
14 |
|
15 |
+
# Interface Gradio simples
|
16 |
interface = gr.Interface(
|
17 |
fn=gerar_sql,
|
18 |
inputs="text",
|
|
|
22 |
)
|
23 |
|
24 |
# Iniciar a interface
|
25 |
+
interface.launch()
|