sourabhbargi11 commited on
Commit
420f5bf
·
verified ·
1 Parent(s): 08ec34d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ def initialize_model():
18
  model = VisionEncoderDecoderModel.from_pretrained("sourabhbargi11/caption4").to(device)
19
  tokenizer = GPT2TokenizerFast.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
20
  image_processor = ViTImageProcessor.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
21
- return image_processor, model,tokenizer, device
22
 
23
  def upload_image():
24
  return st.sidebar.file_uploader("Upload an image (we aren't storing anything)", type=["jpg", "jpeg", "png"])
 
18
  model = VisionEncoderDecoderModel.from_pretrained("sourabhbargi11/caption4").to(device)
19
  tokenizer = GPT2TokenizerFast.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
20
  image_processor = ViTImageProcessor.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
21
+ return image_processor, model,tokenizer, device
22
 
23
  def upload_image():
24
  return st.sidebar.file_uploader("Upload an image (we aren't storing anything)", type=["jpg", "jpeg", "png"])