|
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()) |