Prathamesh1420
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ from huggingface_hub import login
|
|
11 |
# Add your Hugging Face token here
|
12 |
HUGGINGFACE_TOKEN = os.getenv("HF")
|
13 |
login(token=HUGGINGFACE_TOKEN)
|
14 |
-
|
15 |
# Initialize the text classifier for guardrails
|
16 |
classifier = pipeline("text-classification", model="meta-llama/Prompt-Guard-86M")
|
17 |
|
@@ -29,7 +29,7 @@ st.markdown("<h1 style='text-align: center;'>How can I assist you? </h1>", unsaf
|
|
29 |
|
30 |
# Sidebar configuration
|
31 |
st.sidebar.title("😎")
|
32 |
-
|
33 |
summarise_button = st.sidebar.button("Summarise the conversation", key="summarise")
|
34 |
if summarise_button:
|
35 |
st.sidebar.write("Nice chatting with you my friend ❤️:\n\n" + st.session_state['conversation'].memory.buffer)
|
|
|
11 |
# Add your Hugging Face token here
|
12 |
HUGGINGFACE_TOKEN = os.getenv("HF")
|
13 |
login(token=HUGGINGFACE_TOKEN)
|
14 |
+
GROQ_API_KEY = os.getenv("GROQ_API_KEY")
|
15 |
# Initialize the text classifier for guardrails
|
16 |
classifier = pipeline("text-classification", model="meta-llama/Prompt-Guard-86M")
|
17 |
|
|
|
29 |
|
30 |
# Sidebar configuration
|
31 |
st.sidebar.title("😎")
|
32 |
+
|
33 |
summarise_button = st.sidebar.button("Summarise the conversation", key="summarise")
|
34 |
if summarise_button:
|
35 |
st.sidebar.write("Nice chatting with you my friend ❤️:\n\n" + st.session_state['conversation'].memory.buffer)
|