Titobsala commited on
Commit
c350c7c
·
1 Parent(s): 07b8078

config inicial para Streamlite space

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -1,11 +1,17 @@
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
- # Substitua 'seu-modelo' pelo nome ou caminho do seu modelo de geração de texto
5
- gerador = pipeline('text-generation', model='/mnt/c/Users/titob/OneDrive - Ensino Lusófona/EXO/exo-hf/t5-small-60M-esg-keyword/trainig-model/model.safetensors')
 
 
 
 
6
 
7
  st.title('Gerador de Texto')
8
 
 
 
9
  # Entrada do usuário
10
  prompt = st.text_area('Digite o prompt inicial:')
11
 
 
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
+ # Substitua 'seu-nome-de-usuário/nome-do-modelo' pelo caminho correto do seu modelo no Hugging Face Hub
5
+ MODEL_NAME = "exo-is/t5-small-60M-esg-keyword"
6
+
7
+ @st.cache_resource
8
+ def load_model():
9
+ return pipeline('text-generation', model=MODEL_NAME)
10
 
11
  st.title('Gerador de Texto')
12
 
13
+ gerador = load_model()
14
+
15
  # Entrada do usuário
16
  prompt = st.text_area('Digite o prompt inicial:')
17