teste3 / app.py
Elleres's picture
app.py
89e1f13 verified
raw
history blame contribute delete
835 Bytes
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())