Spaces:
Sleeping
Sleeping
Commit
·
5e67c6e
1
Parent(s):
b2a9bc3
test
Browse files
app.py
CHANGED
@@ -48,7 +48,7 @@ def read_pdf(file_path):
|
|
48 |
return output
|
49 |
|
50 |
# Function to query Hugging Face endpoint
|
51 |
-
|
52 |
def query_huggingface(text):
|
53 |
print(zero.device)
|
54 |
tokenizer = AutoTokenizer.from_pretrained("google/gemma-2-9b-it")
|
|
|
48 |
return output
|
49 |
|
50 |
# Function to query Hugging Face endpoint
|
51 |
+
#@spaces.GPU
|
52 |
def query_huggingface(text):
|
53 |
print(zero.device)
|
54 |
tokenizer = AutoTokenizer.from_pretrained("google/gemma-2-9b-it")
|