test

app.py

@@ -2,7 +2,8 @@ from PIL import Image
 import pytesseract
 import os
 import pymupdf
-
+import spaces
+import torch
 import gradio as gr
 from prepare import prepare
 
@@ -20,7 +21,7 @@ from langchain_core.messages import AIMessage, HumanMessage
 from langchain_community.llms import HuggingFaceEndpoint
 from dotenv import load_dotenv
 
-
+zero = torch.Tensor([0]).cuda()
 
 
 def read_pdf(file_path):
@@ -47,7 +48,9 @@ def read_pdf(file_path):
     return output
 
 # Function to query Hugging Face endpoint
+@spaces.GPU
 def query_huggingface(text):
+    print(zero.device)
     load_dotenv()
     api_token = os.getenv("API_TOKEN")
     repo_id = "google/gemma-2-9b-it"