Spaces:

cdcvd
/

Evalution_resume

Sleeping

cdcvd commited on Aug 20

Commit

cd9c591

•

1 Parent(s): 611ad83

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ gemma_pipe = pipeline(
     "text-generation",
     model="google/gemma-2-27b-it",
     model_kwargs={"torch_dtype": torch.bfloat16},
-    device="cuda"
 )
 def extract_text_from_pdf(pdf_file):

     "text-generation",
     model="google/gemma-2-27b-it",
     model_kwargs={"torch_dtype": torch.bfloat16},
+    device="cpu"
 )
 def extract_text_from_pdf(pdf_file):