Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
| 6 |
-
|
| 7 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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):
|