Update app.py

app.py

@@ -14,7 +14,7 @@ api_key = os.getenv("HF_TOKEN")
 client = InferenceClient(api_key=api_key)
 
 # Streamlit app title
-st.title("Text-generation model using Streamlit from Inference API (serverless) feature.") 
+st.title("LM using HF Inference API (serverless) feature.") 
 
 # Ensure the timing variables are initialized in session state
 if "elapsed_time" not in st.session_state:
@@ -25,7 +25,7 @@ if "full_text" not in st.session_state:
     st.session_state["full_text"] = ""
 
 # Model selection dropdown
-model_options = [ "TinyLlama/TinyLlama-1.1B-Chat-v1.0","Qwen/Qwen2.5-1.5B-Instruct", "Qwen/Qwen2.5-72B-Instruct", "meta-llama/Llama-3.2-3B-Instruct"]
+model_options = [ "TinyLlama/TinyLlama-1.1B-Chat-v1.0","Qwen/Qwen2.5-1.5B-Instruct", "Qwen/Qwen2.5-72B-Instruct", "meta-llama/Llama-3.2-3B-Instruct","meta-llama/Llama-3.1-8B-Instruct","meta-llama/Llama-3.2-1B-Instruct"]
 selected_model = st.selectbox("Choose a model:", model_options)
 
 # Create a text input area for user prompts