Correct using state in new gradio version

app.py

@@ -3,13 +3,13 @@ import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from util_funcs import getLengthParam, calcAnswerLengthByProbability, cropContext
 
-def chat_function(Message):   # model, tokenizer
+def chat_function(Message, History):   # model, tokenizer
 
     input_user = Message
 
-    history = gr.get_state() or []
+    History = History or []
 
-    chat_history_ids = torch.zeros((1, 0), dtype=torch.int) if history == [] else torch.tensor(history[-1][2], dtype=torch.long)
+    chat_history_ids = torch.zeros((1, 0), dtype=torch.int) if History == [] else torch.tensor(history[-1][2], dtype=torch.long)
 
     # encode the new user input, add parameters and return a tensor in Pytorch
     lengthId = getLengthParam(input_user, tokenizer)
@@ -78,8 +78,7 @@ def chat_function(Message):   # model, tokenizer
         # Случай когда надо перегенерировать ответ наступил, берем изначальный тензор
         chat_history_ids = chat_history_ids_initial
 
-    history.append((input_user, answer, chat_history_ids.tolist()))        
-    gr.set_state(history)
+    History.append((input_user, answer, chat_history_ids.tolist()))        
     html = "<div class='chatbot'>"
     for user_msg, resp_msg, _ in history:
         if user_msg != '-':
@@ -87,7 +86,7 @@ def chat_function(Message):   # model, tokenizer
         if resp_msg != '-':
             html += f"<div class='resp_msg'>{resp_msg}</div>"
     html += "</div>"
-    return html
+    return html, History
 
 # Download checkpoint:
 
@@ -106,8 +105,8 @@ description = """
 article = "<p style='text-align: center'><a href='https://huggingface.co/avorozhko/ruDialoGpt3-medium-finetuned-context'>Бот на основе дообученной GPT-3</a></p>"
 
 iface = gr.Interface(fn=chat_function,
-                     inputs=gr.inputs.Textbox(lines=3, placeholder="Что вы хотите сказать боту..."),
-                     outputs="html",
+                     inputs=[gr.inputs.Textbox(lines=3, placeholder="Что вы хотите сказать боту..."), 'state'],
+                     outputs=['html', 'state'],
                      title=title, description=description, article=article,
                      theme='dark-grass',
                      css= """
@@ -119,7 +118,7 @@ iface = gr.Interface(fn=chat_function,
                             .panels.unaligned :last-child {order: -1 !important;}
                           """,
                      allow_screenshot=False,
-                     allow_flagging='auto'
+                     allow_flagging='never'
                     )
 
 if __name__ == "__main__":

requirements.txt

@@ -1,3 +1,2 @@
 transformers
-torch
-gradio==2.7.0
+torch