File size: 784 Bytes
6ddeb4a
 
0aef548
 
6ddeb4a
 
 
 
 
 
 
 
 
 
 
0aef548
 
 
 
 
 
6ddeb4a
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
import streamlit as st
from langchain.llms import HuggingFaceHub
import os
from langchain.llms import HuggingFaceHub

st.set_page_config(page_title="Educational Chatbot")

st.title("πŸŽ“ Educational Chatbot")

user_input = st.text_input("Ask me anything:")

if "history" not in st.session_state:
    st.session_state.history = []

if user_input:
    llm = HuggingFaceHub(
    repo_id="tiiuae/falcon-7b-instruct",
    huggingfacehub_api_token=os.environ.get("Educational_chatbot"),
    model_kwargs={"temperature": 0.5, "max_new_tokens": 100}
)

    response = llm(user_input)
    st.session_state.history.append(("You", user_input))
    st.session_state.history.append(("Bot", response))

for sender, msg in reversed(st.session_state.history):
    st.markdown(f"**{sender}:** {msg}")