fixed missing import

This commit is contained in:
JamePeng
2024-08-27 20:52:52 +08:00
committed by GitHub
parent 344ddc2cb1
commit b109c67478

View File

@@ -4,6 +4,7 @@ import streamlit as st
import torch import torch
from PIL import Image from PIL import Image
from decord import VideoReader, cpu from decord import VideoReader, cpu
import numpy as np
from transformers import AutoModel, AutoTokenizer from transformers import AutoModel, AutoTokenizer
# Model path # Model path