File size: 561 Bytes
19f4fce |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
import os
import time
import openai
import streamlit as st
from models.llamaCustom import LlamaCustom
from utils.chatbox import *
st.set_page_config(page_title="Llama", page_icon="🦙")
st.subheader("Llama Index with Custom LLM Demo")
if "messages" not in st.session_state:
st.session_state.messages = []
if "openai_api_key" not in st.session_state:
st.info("Enter your openai key to access the chatbot.")
else:
with st.spinner("Initializing vector index"):
model = LlamaCustom(name="llamaCustom")
chatbox("llama_custom", model)
|