Spaces:
Running
on
Zero
Running
on
Zero
Update main.py
Browse files
main.py
CHANGED
@@ -10,6 +10,9 @@ import spaces # for GPU
|
|
10 |
import threading
|
11 |
import time
|
12 |
|
|
|
|
|
|
|
13 |
# Load the dataset and convert to pandas
|
14 |
full_data = datasets.load_dataset("ccm/publications")["train"].to_pandas()
|
15 |
|
@@ -128,7 +131,7 @@ gradio.ChatInterface(
|
|
128 |
chatbot = gradio.Chatbot(
|
129 |
show_label=False,
|
130 |
show_copy_button=True,
|
131 |
-
value=[None,
|
132 |
),
|
133 |
retry_btn = None,
|
134 |
undo_btn = None,
|
|
|
10 |
import threading
|
11 |
import time
|
12 |
|
13 |
+
# Constants
|
14 |
+
GREETING = "Hi there! I'm an AI agent that answers your questions using a [retrieval-augmented generation]() pipeline that searches over publications abstracts from the Design Research Collective."
|
15 |
+
|
16 |
# Load the dataset and convert to pandas
|
17 |
full_data = datasets.load_dataset("ccm/publications")["train"].to_pandas()
|
18 |
|
|
|
131 |
chatbot = gradio.Chatbot(
|
132 |
show_label=False,
|
133 |
show_copy_button=True,
|
134 |
+
value=[None, GREETING]
|
135 |
),
|
136 |
retry_btn = None,
|
137 |
undo_btn = None,
|