daniel05 commited on
Commit
dfae60a
·
verified ·
1 Parent(s): 0431336

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -2,9 +2,13 @@ import streamlit as st
2
  import requests
3
  from transformers import T5Tokenizer, T5ForConditionalGeneration
4
  # URL da API (substitua pelo seu endpoint de API, ex: ngrok)
5
- model_name = "Text-to-SQL T5 modelo 1" # Você pode trocar para o modelo que está usando, como "t5-base", "t5-large", etc.
6
- tokenizer = T5Tokenizer.from_pretrained(model_name)
7
- model = T5ForConditionalGeneration.from_pretrained(model_name)
 
 
 
 
8
  st.set_page_config(layout="wide")
9
  st.title("Text-to-SQL app")
10
  def preditor_sql(texto):
 
2
  import requests
3
  from transformers import T5Tokenizer, T5ForConditionalGeneration
4
  # URL da API (substitua pelo seu endpoint de API, ex: ngrok)
5
+ @st.cache_data
6
+ def model1()
7
+ model_name = "Text-to-SQL T5 modelo 1" # Você pode trocar para o modelo que está usando, como "t5-base", "t5-large", etc.
8
+ tokenizer = T5Tokenizer.from_pretrained(model_name)
9
+ model = T5ForConditionalGeneration.from_pretrained(model_name)
10
+ return model,tokenizer
11
+ model,tokenizer = model1()
12
  st.set_page_config(layout="wide")
13
  st.title("Text-to-SQL app")
14
  def preditor_sql(texto):