Update app.py

app.py

@@ -2,9 +2,16 @@ import streamlit as st
 from transformers import T5Tokenizer, T5ForConditionalGeneration
 import huggingface_hub
 import torch
+import os
+
+# Load Hugging Face API key from environment variable
+HF_API_KEY = os.getenv("HF_API_KEY")
+if not HF_API_KEY:
+    st.error("Please set your Hugging Face API key in the environment variables.")
+    st.stop()
 
 # Login to Hugging Face Hub
-# huggingface_hub.login("")
+huggingface_hub.login(HF_API_KEY)
 
 # Load model and tokenizer
 repo_id = "Darshan03/t5-model-small"  # Replace with your hub username and model name