shadowlilac commited on
Commit
58ef361
1 Parent(s): daa9e93

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,8 +2,8 @@ import gradio as gr
2
  from PIL import Image
3
  from transformers import AutoProcessor, BlipForConditionalGeneration
4
  import os
5
- processor = AutoProcessor.from_pretrained("shadowlilac/visor", token=os.getenv('HUGGING_FACE_HUB_TOKEN'))
6
- model = BlipForConditionalGeneration.from_pretrained("shadowlilac/visor", ignore_mismatched_sizes=True, token=os.getenv('HUGGING_FACE_HUB_TOKEN'))
7
 
8
  def generate_caption(image):
9
  raw_image = Image.fromarray(image)
 
2
  from PIL import Image
3
  from transformers import AutoProcessor, BlipForConditionalGeneration
4
  import os
5
+ processor = AutoProcessor.from_pretrained("shadowlilac/visor")
6
+ model = BlipForConditionalGeneration.from_pretrained("shadowlilac/visor", ignore_mismatched_sizes=True)
7
 
8
  def generate_caption(image):
9
  raw_image = Image.fromarray(image)