test

app.py

@@ -23,7 +23,8 @@ from dotenv import load_dotenv
 from huggingface_hub import InferenceApi
 
 #zero = torch.Tensor([0]).cuda()
-
+load_dotenv()
+api_token = os.getenv("HF_TOKEN")
 #@spaces.GPU
 def read_pdf(file_path):
     output = ''
@@ -51,7 +52,7 @@ def read_pdf(file_path):
 # Function to query Hugging Face endpoint
 #@spaces.GPU
 def query_huggingface(text):
-    api = InferenceApi(repo_id="google/gemma-2-9b-it", token="your_token_here")
+    api = InferenceApi(repo_id="google/gemma-2-9b-it", token=api_token)
     response = api(inputs=text)
     return response