Update app.py

app.py

@@ -15,14 +15,17 @@ model_id = st.selectbox(
     "Select a LLaMA model",
     ["meta-llama/Llama-3.2-3B-Instruct", "meta-llama/Llama-3.2-1B-Instruct"]  # Replace with the correct model names
 )
-client = InferenceClient(model=model_id,token=hf_token)
+
+# Initialize the Inference Client
+client = InferenceClient(token=hf_token)
 
 # Button to trigger the inference
 if st.button("Generate Text"):
     if user_input:
         with st.spinner(f"Generating text using {model_id}..."):
             # Perform inference using the selected model
-            response = client.text_generation(model=
+            response = client.text_generation(
+                model=model_id,
                 messages=[
                     {"role": "system", "content": "You are a question answering assistant."},
                     {"role": "user", "content": user_input}
@@ -35,3 +38,5 @@ if st.button("Generate Text"):
     else:
         st.warning("Please enter a prompt to generate text.")
 
+
+