Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
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"])
|