| import streamlit as st |
| from transformers import AutoModelForSequenceClassification, AutoTokenizer |
|
|
| |
| modelo = AutoModelForSequenceClassification.from_pretrained("nome-do-modelo") |
| tokenizer = AutoTokenizer.from_pretrained("nome-do-modelo") |
|
|
| |
| st.title("Assistente Virtual Humanizado") |
| st.write("Olá! Como posso ajudá-lo hoje?") |
|
|
| |
| pergunta = st.text_input("Insira sua pergunta") |
|
|
| |
| if st.button("Enviar"): |
| |
| inputs = tokenizer(pergunta, return_tensors="pt") |
| outputs = modelo(**inputs) |
| resposta = outputs.logits.argmax(-1) |
| |
| |
| st.write("Resposta:", resposta.item()) |