Update app.py

app.py

@@ -2,26 +2,39 @@ import os
 import torch
 import streamlit as st
 from transformers import pipeline
+from huggingface_hub import login
 
 # Access Hugging Face API token from environment variables
 api_key = os.getenv("llama3")
 
-# Check if the token is provided (for debugging)
 if not api_key:
     st.error("Hugging Face API token is missing!")
+    st.stop()  # Stop execution if no API key is found
 
-# Load the model from Hugging Face
+# Authenticate with Hugging Face Hub
+try:
+    login(api_key)
+except Exception as e:
+    st.error(f"Authentication failed: {e}")
+    st.stop()
+
+# Load the model
 model_id = "meta-llama/Llama-3.2-3B"
-pipe = pipeline(
-    "text-generation",
-    model=model_id,
-    torch_dtype=torch.bfloat16,
-    device_map="auto",
-)
+
+try:
+    pipe = pipeline(
+        "text-generation",
+        model=model_id,
+        torch_dtype=torch.bfloat16,
+        device_map="auto",
+    )
+except Exception as e:
+    st.error(f"Error loading model: {e}")
+    st.stop()
 
 # Streamlit interface
-st.title("Pirate Chatbot")
-st.write("Ask me anything, and I'll respond in pirate speak!")
+st.title("Pirate Chatbot 🏴‍☠️")
+st.write("Ahoy, matey! Ask me anything, and I'll respond in pirate speak! ☠️⚓")
 
 # User input
 user_input = st.text_input("Your question:", "")
@@ -32,8 +45,8 @@ if user_input:
         {"role": "user", "content": user_input},
     ]
 
-    # Generate response
-    outputs = pipe(messages, max_new_tokens=256)
-
-    # Show the response
-    st.write(outputs[0]["generated_text"][-1])
+    try:
+        outputs = pipe(messages, max_new_tokens=256)
+        st.write(outputs[0]["generated_text"][-1])
+    except Exception as e:
+        st.error(f"Error generating response: {e}")