davidberenstein1957's picture
Update LLM preference collector
17aeee6
raw
history blame
868 Bytes
import random
from chat_interface_preference import ChatInterface
def random_response(message, history, request):
response = random.choice(["Yes", "No"])
response_total = ""
for char in response:
response_total += char
yield response_total
style = "<style>.user-message,.system-message{display:flex;margin:10px}.user-message .message-content{background-color:#c2e3f7;color:#000}.system-message .message-content{background-color:#f5f5f5;color:#000}.message-content{padding:10px;border-radius:10px;max-width:70%;word-wrap:break-word}.container{display:flex;justify-content:space-between}.column{width:48%}</style>"
demo = ChatInterface(
random_response,
cache_examples=False,
css=style,
repo_id="geitje-spin-preference",
prefence_techniques=["kto"],
max_turns=10,
)
if __name__ == "__main__":
demo.launch()