Spaces:
Runtime error
Runtime error
""" | |
# 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/" | |