Hugging Face's logo

Spaces:
triphuong57
/
paligemma_ft_v1
Runtime error

App Files Community
triphuong57 commited on
Commit
540ec9f
·
verified ·
1 Parent(s): 19f033c

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('token')
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=180)
 
8
  token = os.getenv('token')
9
  HfFolder.save_token(token)
10
  device = "cuda"
11
+ model = PaliGemmaForConditionalGeneration.from_pretrained("triphuong57/paligemma_lora").to("cuda")
12
  processor = AutoProcessor.from_pretrained("google/paligemma-3b-mix-224")
13
 
14
  @spaces.GPU(duration=180)