Update app.py

app.py

@@ -14,6 +14,8 @@ from transformers import AutoModelForCausalLM
 from transformers import AutoTokenizer
 import time
 
+convoState = gr.State([""])
+
 EXAMPLE_PROMPT = """This is a tool for helping someone with memory issues remember the next word. 
 The predictions follow a few rules:
 1) The predictions are suggestions of ways to continue the transcript as if someone forgot what the next word was.
@@ -72,14 +74,15 @@ def inference(audio, prompt, model, temperature, latest):
         #print("Infered type is: ", type(infered))
         infers = list(map(lambda x: x.replace("\n", ""), temp))
         #infered = list(map(lambda x: x.split(','), infers))
+        convoState: latest
         
 
-    return transcript, infers, gr.State.convoState: latest
+    return transcript, infers, convoState
 
 
 # get audio from microphone 
 with gr.Blocks() as face:
-    convoState = gr.State([""])
+    
     with gr.Row():
         with gr.Column():
             audio = gr.Audio(source="microphone", type="filepath")