Spaces:
Runtime error
Runtime error
disabled st.cache for the inference
Browse files- app/app.py +2 -1
app/app.py
CHANGED
@@ -35,7 +35,8 @@ def get_generator(model_name: str):
|
|
35 |
text_generator = pipeline('text-generation', model=model_name)
|
36 |
return text_generator
|
37 |
|
38 |
-
|
|
|
39 |
def process(text_generator, text: str, max_length: int = 100, do_sample: bool = True, top_k: int = 50, top_p: float = 0.95,
|
40 |
temperature: float = 1.0, max_time: float = 60.0, seed=42):
|
41 |
# st.write("Cache miss: process")
|
|
|
35 |
text_generator = pipeline('text-generation', model=model_name)
|
36 |
return text_generator
|
37 |
|
38 |
+
# Disable the st.cache for this function due to issue on newer version of streamlit
|
39 |
+
# @st.cache(suppress_st_warning=True, hash_funcs={tokenizers.Tokenizer: id})
|
40 |
def process(text_generator, text: str, max_length: int = 100, do_sample: bool = True, top_k: int = 50, top_p: float = 0.95,
|
41 |
temperature: float = 1.0, max_time: float = 60.0, seed=42):
|
42 |
# st.write("Cache miss: process")
|