🌘w🌖

app.py

@@ -63,7 +63,7 @@ def format_prompt(prompt,retrieved_documents,k):
 
 
 @spaces.GPU(duration=150)
-def talk(message,history):
+def talk(prompt,history):
     k = 1 # number of retrieved documents
     scores , retrieved_documents = search(prompt, k)
     formatted_prompt = format_prompt(prompt,retrieved_documents,k)