Madhuri123 commited on
Commit
f577c76
·
verified ·
1 Parent(s): 5254778

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ st.title("LLM Model Inference with Streamlit")
11
  # Dropdown for selecting between LLaMA 3 models
12
  model_id = st.selectbox(
13
  "Select a LLaMA model",
14
- ["meta-llama/Llama-3.2-3B-Instruct","google/gemma-7b","meta-llama/Llama-3.2-1B-Instruct","meta-llama/Meta-Llama-3-8B-Instruct"] # Replace with the correct model names
15
  )
16
  user_input = st.chat_input("Enter a text prompt:")
17
  # Initialize the Inference Client
 
11
  # Dropdown for selecting between LLaMA 3 models
12
  model_id = st.selectbox(
13
  "Select a LLaMA model",
14
+ ["meta-llama/Llama-3.2-3B-Instruct","meta-llama/Llama-3.2-1B-Instruct","meta-llama/Meta-Llama-3-8B-Instruct"] # Replace with the correct model names
15
  )
16
  user_input = st.chat_input("Enter a text prompt:")
17
  # Initialize the Inference Client