Update app.py

app.py

@@ -1,6 +1,6 @@
 import streamlit as st
 import os
-from transformers import pipeline
+from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
 from huggingface_hub import login
 
 # Read the Hugging Face API token from the environment variable
@@ -14,8 +14,8 @@ login(token=api_token)
 
 model_id = "meta-llama/Meta-Llama-3-8B"
 
-# Initialize the model pipeline
-pipe = pipeline("text-generation", model=model_id)
+# Initialize the model pipeline with authentication
+pipe = pipeline("text-generation", model=model_id, use_auth_token=api_token)
 
 # Display the logo and title
 st.image("logo.jpg", width=300)
@@ -49,3 +49,4 @@ if user_input:
         
         # Display the AI's response
         st.write(ai_response)
+