Update app.py

app.py

@@ -1,17 +1,20 @@
 import os
 import requests
 # from IPython.display import Markdown, display, update_display
-from openai import OpenAI
-from google.colab import drive
+# from openai import OpenAI
+# from google.colab import drive
 from huggingface_hub import login
-from google.colab import userdata
+# from google.colab import userdata
 from transformers import AutoTokenizer, AutoModelForCausalLM, TextStreamer, BitsAndBytesConfig
 import torch
+from dotenv import load_dotenv
+
+load_dotenv()
 
 
 LLAMA = "meta-llama/Meta-Llama-3.1-8B-Instruct"
 
-hf_token = userdata.get('HF_TOKEN')
+hf_token = os.environ["GOOGLE_API_KEY"]
 login(hf_token, add_to_git_credential=True)
 
 system_message = "You are an assistant that produces datasets based on description provided."