Spaces:
Running
Running
from einops import rearrange | |
import gradio as gr | |
import torch | |
import torch.nn.functional as F | |
from PIL import Image, ImageOps | |
from transformers import AutoModel, CLIPImageProcessor | |
hf_repo = "nvidia/RADIO-L" | |
image_processor = CLIPImageProcessor.from_pretrained(hf_repo) | |
model = AutoModel.from_pretrained(hf_repo, trust_remote_code=True) | |
model.eval().cuda() | |