| """ | |
| # Copyright (c) 2022, salesforce.com, inc. | |
| # All rights reserved. | |
| # SPDX-License-Identifier: BSD-3-Clause | |
| # For full license text, see the LICENSE file in the repo root or https://opensource.org/licenses/BSD-3-Clause | |
| """ | |
| from PIL import Image | |
| import requests | |
| import streamlit as st | |
| import torch | |
| def load_demo_image(): | |
| img_url = ( | |
| "https://storage.googleapis.com/sfr-vision-language-research/BLIP/demo.jpg" | |
| ) | |
| raw_image = Image.open(requests.get(img_url, stream=True).raw).convert("RGB") | |
| return raw_image | |
| device = torch.device("cuda" if torch.cuda.is_available() else "cpu") | |
| cache_root = "/export/home/.cache/lavis/" | |