Spaces:

shamim237
/

summarize

Runtime error

shamim237 commited on Jan 26, 2023

Commit

771a4bb

•

1 Parent(s): 5f5a6ec

Update summary.py

Files changed (1) hide show

summary.py CHANGED Viewed

@@ -7,7 +7,7 @@ device = 'cuda' if torch.cuda.is_available() else 'cpu'
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = PegasusForConditionalGeneration.from_pretrained(model_name).to(device)
-@st.cache(allow_output_mutation=True)
 def summarize(passage):
     txt = " ".join(passage)

 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = PegasusForConditionalGeneration.from_pretrained(model_name).to(device)
+# @st.cache(allow_output_mutation=True)
 def summarize(passage):
     txt = " ".join(passage)