cosmo3769 commited on
Commit
194b786
·
verified ·
1 Parent(s): 53e219f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -1,8 +1,11 @@
1
  import gradio as gr
2
  from transformers import AutoProcessor, PaliGemmaForConditionalGeneration
 
3
  from PIL import Image
4
  import torch
5
 
 
 
6
  # Load model and processor
7
  model_id = "cosmo3769/finetuned_paligemma_vqav2_small"
8
  model = PaliGemmaForConditionalGeneration.from_pretrained(model_id)
 
1
  import gradio as gr
2
  from transformers import AutoProcessor, PaliGemmaForConditionalGeneration
3
+ from huggingface_hub import login
4
  from PIL import Image
5
  import torch
6
 
7
+ login()
8
+
9
  # Load model and processor
10
  model_id = "cosmo3769/finetuned_paligemma_vqav2_small"
11
  model = PaliGemmaForConditionalGeneration.from_pretrained(model_id)