🌘w🌖

app.py

@@ -50,7 +50,7 @@ def prepare_prompt(query, retrieved_examples):
     titles = titles[::-1]
     for i in range(TOP_K):
         prompt += f"Title: {titles[i]}, Text: {texts[i]}\n"
-    return prompt, (titles, urls)
+    return prompt, zip(titles, urls)
 
 
 @spaces.GPU(duration=150)
@@ -93,9 +93,9 @@ def talk(message, history):
     for new_text in streamer:
         partial_text += new_text
         print("partial_text : ", partial_text)
-        # yield partial_text
+        yield partial_text
     partial_text += resources
-    return partial_text
+    yield partial_text
 
 
 TITLE = "# RAG"