Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
import streamlit as st
|
2 |
from transformers import (
|
3 |
AutoTokenizer,
|
4 |
-
AutoModelForCausalLM,
|
5 |
TextIteratorStreamer
|
6 |
)
|
7 |
from huggingface_hub import login
|
@@ -97,7 +97,7 @@ if prompt := st.chat_input("Faca uma pergunta sobre a LGT..."):
|
|
97 |
|
98 |
# Load model if not loaded
|
99 |
if "model" not in st.session_state:
|
100 |
-
with st.spinner("π A carregar o modelo ..."):
|
101 |
model, tokenizer = load_model()
|
102 |
if not model:
|
103 |
st.stop()
|
|
|
1 |
import streamlit as st
|
2 |
from transformers import (
|
3 |
AutoTokenizer,
|
4 |
+
AutoModelForCausalLM,
|
5 |
TextIteratorStreamer
|
6 |
)
|
7 |
from huggingface_hub import login
|
|
|
97 |
|
98 |
# Load model if not loaded
|
99 |
if "model" not in st.session_state:
|
100 |
+
with st.spinner("π A carregar o modelo PT-T5..."):
|
101 |
model, tokenizer = load_model()
|
102 |
if not model:
|
103 |
st.stop()
|