| """
|
| # 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
|
|
|
|
|
| @st.cache()
|
| 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/"
|
|
|
