Update app.py

app.py

@@ -1,6 +1,5 @@
 import os
-from transformers import pipeline, AutoTokenizer, AutoModelForTextGeneration
-
+from transformers import pipeline
 # Retrieve the API token from environment variables
 api_token = os.getenv('hugging_face_api_token')
 
@@ -11,10 +10,6 @@ if not api_token:
 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="google/gemma-7b", tokenizer="google/gemma-7b")