Hugging Face's logo

Spaces:
triphuong57
/
paligemma_finetune_v2
Runtime error

App Files Community
triphuong57 commited on
Commit
7bcf258
·
verified ·
1 Parent(s): 3283fba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ import os
8
  token = os.getenv('toke')
9
  HfFolder.save_token(token)
10
  device = "cuda"
11
- model = PaliGemmaForConditionalGeneration.from_pretrained("triphuong57/paligemma_lora")
12
  processor = AutoProcessor.from_pretrained("google/paligemma-3b-mix-224")
13
 
14
  @spaces.GPU(duration=120)
 
8
  token = os.getenv('toke')
9
  HfFolder.save_token(token)
10
  device = "cuda"
11
+ model = PaliGemmaForConditionalGeneration.from_pretrained("triphuong57/paligemma_lora").to(device)
12
  processor = AutoProcessor.from_pretrained("google/paligemma-3b-mix-224")
13
 
14
  @spaces.GPU(duration=120)