fix

app.py

@@ -196,7 +196,7 @@ def load_sampler(cls_model_name, lm_tokenizer):
         return sampler
 
 
-@st.cache
+@st.cache(suppress_st_warning=True)
 def inference(
         lm_model_name: str,
         cls_model_name: str,

generator.py

@@ -44,7 +44,7 @@ class Generator:
         caif_period: int,
         caif_tokens_num: Union[int, None] = None,
         entropy: float = None,
-        progress_bar = None,
+        progress_bar=None,
         **sampler_kwargs
     ):
         self.entropy = entropy
@@ -66,7 +66,7 @@ class Generator:
                 caif_tokens_num=caif_tokens_num,
                 **sampler_kwargs
             )
-            progress_bar.step((i+1)/max_length)
+            progress_bar.progress((i+1)/max_length)
             if ended_sequences.all():
                 break