Hugging Face's logo

Spaces:
triphuong57
/
paligemma_ft_v1
Runtime error

App Files Community
triphuong57 commited on
Commit
52d6487
·
verified ·
1 Parent(s): f52def6

Update app.py

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