Ryu-m0m commited on
Commit
2ddb416
β€’
1 Parent(s): 1f4d028

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -21,7 +21,7 @@ import string
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
 
27
  # Hugging Face login
@@ -125,7 +125,7 @@ def rag_advanced(user_query):
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: ", "")
 
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 = 'mistralai/Mistral-7B-Instruct-v0.3'
25
  HF_TOKEN = st.sidebar.text_input("Enter your Hugging Face token:", "YOUR_HF_TOKEN", type="password")
26
 
27
  # Hugging Face login
 
125
 
126
  # Streamlit UI
127
  st.title("Wikipedia RAG Chatbot")
128
+ st.markdown("Choose a topic. 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: ", "")