Spaces:
Sleeping
Sleeping
| 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()) |