Hugging Face's logo

Spaces:
xelpmocAI
/
PaySlip_Demo
Paused

App Files Community
xelpmocAI commited on
Commit
2fecabe
·
verified ·
1 Parent(s): ec4bd6c

Removed manual to device to let accelerator handle it

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -13,8 +13,7 @@ model = Qwen2VLForConditionalGeneration.from_pretrained(
13
 
14
  # default processer
15
  processor = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-7B-Instruct")
16
- device = "cuda" if torch.cuda.is_available() else "cpu"
17
- model.to(device)
18
 
19
 
20
  other_benifits = '''Extract the following information in the given format:
 
13
 
14
  # default processer
15
  processor = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-7B-Instruct")
16
+
 
17
 
18
 
19
  other_benifits = '''Extract the following information in the given format: