Ryu-m0m commited on
Commit
1f4d028
·
verified ·
1 Parent(s): 80fb749

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -19,8 +19,8 @@ import random
19
  import string
20
 
21
  # User variables
22
- topic = st.sidebar.text_input("Enter the Wikipedia topic:", "Wikipedia_Keyword")
23
- query = st.sidebar.text_input("Enter your first query:", "First query")
24
  model_name = st.sidebar.selectbox("Select model:", ["mistralai/Mistral-7B-Instruct-v0.3", "meta-llama/Meta-Llama-3.1-8B-Instruct"])
25
  HF_TOKEN = st.sidebar.text_input("Enter your Hugging Face token:", "YOUR_HF_TOKEN", type="password")
26
 
@@ -125,6 +125,7 @@ def rag_advanced(user_query):
125
 
126
  # Streamlit UI
127
  st.title("Wikipedia RAG Chatbot")
 
128
 
129
  # Input box for the user to type their message
130
  user_input = st.text_input("You: ", "")
 
19
  import string
20
 
21
  # User variables
22
+ topic = st.sidebar.text_input("Enter the Wikipedia topic:", "Japanese History")
23
+ #query = st.sidebar.text_input("Enter your first query:", "First query")
24
  model_name = st.sidebar.selectbox("Select model:", ["mistralai/Mistral-7B-Instruct-v0.3", "meta-llama/Meta-Llama-3.1-8B-Instruct"])
25
  HF_TOKEN = st.sidebar.text_input("Enter your Hugging Face token:", "YOUR_HF_TOKEN", type="password")
26
 
 
125
 
126
  # Streamlit UI
127
  st.title("Wikipedia RAG Chatbot")
128
+ st.markdown("Choose topic and model. Don't forget to put your 🤗 token!")
129
 
130
  # Input box for the user to type their message
131
  user_input = st.text_input("You: ", "")