File size: 835 Bytes
89e1f13 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
import streamlit as st
from transformers import AutoModelForSequenceClassification, AutoTokenizer
# Carregue o modelo de IA
modelo = AutoModelForSequenceClassification.from_pretrained("nome-do-modelo")
tokenizer = AutoTokenizer.from_pretrained("nome-do-modelo")
# Crie a interface de usuário
st.title("Assistente Virtual Humanizado")
st.write("Olá! Como posso ajudá-lo hoje?")
# Crie um campo de texto para o usuário inserir sua pergunta
pergunta = st.text_input("Insira sua pergunta")
# Crie um botão para o usuário enviar sua pergunta
if st.button("Enviar"):
# Processa a pergunta utilizando o modelo de IA
inputs = tokenizer(pergunta, return_tensors="pt")
outputs = modelo(**inputs)
resposta = outputs.logits.argmax(-1)
# Exiba a resposta para o usuário
st.write("Resposta:", resposta.item()) |