BugZoid commited on
Commit
b564db7
·
verified ·
1 Parent(s): fcb0322

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
- AutoModelForSeq2SeqGeneration,
5
  T5ForConditionalGeneration,
6
  T5Tokenizer
7
  )
@@ -14,7 +14,7 @@ if 'models_loaded' not in st.session_state:
14
 
15
  # Load the paraphrasing model and tokenizer
16
  st.session_state.paraphrase_tokenizer = AutoTokenizer.from_pretrained("facebook/bart-large-cnn")
17
- st.session_state.paraphrase_model = AutoModelForSeq2SeqGeneration.from_pretrained("facebook/bart-large-cnn")
18
 
19
  st.session_state.models_loaded = True
20
 
 
1
  import streamlit as st
2
  from transformers import (
3
  AutoTokenizer,
4
+ AutoModelForSeq2SeqLM, # Correção aqui
5
  T5ForConditionalGeneration,
6
  T5Tokenizer
7
  )
 
14
 
15
  # Load the paraphrasing model and tokenizer
16
  st.session_state.paraphrase_tokenizer = AutoTokenizer.from_pretrained("facebook/bart-large-cnn")
17
+ st.session_state.paraphrase_model = AutoModelForSeq2SeqLM.from_pretrained("facebook/bart-large-cnn")
18
 
19
  st.session_state.models_loaded = True
20