amiguel commited on
Commit
211d6b8
Β·
verified Β·
1 Parent(s): 1dfd2ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import streamlit as st
2
  from transformers import (
3
  AutoTokenizer,
4
- AutoModelForCausalLM, # Use AutoModelForCausalLM instead of AutoModelForSeq2SeqLM
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()