huggingface-transformers - 当我使用 st.cache 时 Streamlit Unhashable TypeError

Question

当我使用 st.cache 装饰器来兑现拥抱面变压器模型时，我得到

Unhashable TypeError

这是代码

from transformers import pipeline 
import streamlit as st 
from io import StringIO

@st.cache(hash_funcs={StringIO: StringIO.getvalue})
def model() :
    return pipeline("sentiment-analysis", model='akhooli/xlm-r-large-arabic-sent')

Answer 1

在 Streamlit 存储库的问题部分搜索后

我发现散列参数不是必需的，只需要传递这个参数

allow_output_mutation = True