Spaces:
Sleeping
Sleeping
import streamlit as st | |
from tensorflow.keras.models import load_model | |
import numpy as np | |
from transformers import logging | |
# Hugging Face Transformers kütüphanesinden gelen hataları kapat | |
logging.set_verbosity_error() | |
# Streamlit başlık | |
st.title('Chat with Me') | |
# Model yükleme (burada dil modeli veya sohbet modeli olmalı) | |
model = load_model('text_classification_model.h5') | |
# Chat history | |
if 'chat_history' not in st.session_state: | |
st.session_state.chat_history = [] | |
# Kullanıcıdan giriş alma | |
soru = st.text_input('Sor:') | |
# Yanıt üretme fonksiyonu (modelinize göre bu kısım değişebilir) | |
def generate_response(input_text): | |
# Örneğin, modelinize uygun şekilde giriş ve çıkışları dönüştürün | |
# Bu basit bir örnek, modelinizin gereksinimlerine göre düzenlemelisiniz | |
input_vector = np.array([input_text]) # Modelin beklediği giriş formatı | |
prediction = model.predict(input_vector) | |
response = "Bu bir örnek yanıttır." # Modelden gelen yanıtı buraya yazın | |
return response | |
# "Sor" butonuna tıklama olayını işleme | |
if st.button('Sor'): | |
if soru: | |
response = generate_response(soru) | |
st.session_state.chat_history.append({'role': 'user', 'text': soru}) | |
st.session_state.chat_history.append({'role': 'model', 'text': response}) | |
st.experimental_rerun() | |
# Sohbet geçmişini gösterme | |
for message in reversed(st.session_state.chat_history): | |
if message['role'] == 'user': | |
st.markdown(f'<div style="text-align: right; background-color: #2F2F2F; padding: 10px; border-radius: 10px; margin: 10px; width: fit-content;">👤 Sen: {message["text"]}</div>', unsafe_allow_html=True) | |
elif message['role'] == 'model': | |
st.markdown(f'<div style="text-align: left; background-color: #2E2E2E; padding: 10px; border-radius: 10px; margin: 10px; width: fit-content;">🤖 Bot: {message["text"]}</div>', unsafe_allow_html=True) | |
# "Yeni Sohbet" butonuna tıklama olayını işleme | |
if st.button('Yeni Sohbet'): | |
st.session_state.chat_history = [] | |
st.experimental_rerun() | |