JeffersonCorreiax commited on
Commit
42ba2ac
·
verified ·
1 Parent(s): d5ddec9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -8
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()