Prathamesh1420 commited on
Commit
469b8a3
·
verified ·
1 Parent(s): 01993f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- groq_api_key = st.sidebar.text_input(label="Groq API Key", type="password")
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)