Update app.py

app.py

@@ -1,17 +1,26 @@
-import streamlit as st
 import os
-from transformers import pipeline
+from transformers import pipeline, AutoTokenizer, AutoModelForTextGeneration
 
-# Set up the Hugging Face API token
+# Retrieve the API token from environment variables
+api_token = os.getenv('hugging_face_api_token')
+
+if not api_token:
+    raise ValueError("Hugging Face API token not found. Please set the HUGGING_FACE_API_TOKEN environment variable.")
+
+# Configure the use of the token for Hugging Face operations
+from huggingface_hub import HfFolder
+HfFolder.save_token(api_token)
+
+# Now you can securely use the model
+model_name = "google/gemma-7b"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForTextGeneration.from_pretrained(model_name)
+
+generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
-hugging_face_api_token = os.getenv('hugging_face_api_token')
 
-# Load the text generation pipeline
-generator = pipeline("text-generation", model="google/gemma-7b", tokenizer="google/gemma-7b")
 
 
-import streamlit as st
-from transformers import pipeline
 
 # Streamlit app title and description
 st.title("Gemma Text Generation App")