Update app.py

app.py

@@ -76,7 +76,7 @@ def set_past_key_values():
     
 
 @spaces.GPU
-def generate(text, past_key_values):
+def generate(text):
     messages = [
         *start_messages,
         {'role': 'user', 'content': text}
@@ -90,7 +90,7 @@ def generate(text, past_key_values):
 if __name__ == "__main__":
     with torch.no_grad():
         set_past_key_values()
-        print(past_key_values)
-        demo = gr.Interface(partial(generate), 
+        print(f'{past_key_values.past_key_values}')
+        demo = gr.Interface(generate, 
                             inputs="textbox", outputs="textbox")
         demo.launch()