Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -76,7 +76,6 @@ def predict_chat(message: str, history: list):
|
|
76 |
return
|
77 |
|
78 |
# history is already in the 'messages' format if type='messages' is set on chatbot
|
79 |
-
# It contains dictionaries with 'role' and 'content'
|
80 |
messages = [{"role": "system", "content": "You are a friendly chatbot."}] + history
|
81 |
messages.append({"role": "user", "content": message})
|
82 |
|
@@ -131,15 +130,18 @@ if __name__ == "__main__":
|
|
131 |
load_model_for_zerocpu()
|
132 |
|
133 |
# Initial message for the chatbot in the 'messages' format
|
134 |
-
|
135 |
"Hello! I'm an AI assistant. I'm currently running in a CPU-only "
|
136 |
"environment for efficient demonstration. How can I help you today?"
|
137 |
}]
|
138 |
|
|
|
|
|
|
|
139 |
demo = gr.ChatInterface(
|
140 |
fn=predict_chat,
|
141 |
-
#
|
142 |
-
chatbot=
|
143 |
textbox=gr.Textbox(
|
144 |
placeholder="Ask me a question...",
|
145 |
container=False,
|
@@ -162,4 +164,8 @@ if __name__ == "__main__":
|
|
162 |
clear_btn="Clear Chat"
|
163 |
)
|
164 |
|
|
|
|
|
|
|
|
|
165 |
demo.launch()
|
|
|
76 |
return
|
77 |
|
78 |
# history is already in the 'messages' format if type='messages' is set on chatbot
|
|
|
79 |
messages = [{"role": "system", "content": "You are a friendly chatbot."}] + history
|
80 |
messages.append({"role": "user", "content": message})
|
81 |
|
|
|
130 |
load_model_for_zerocpu()
|
131 |
|
132 |
# Initial message for the chatbot in the 'messages' format
|
133 |
+
initial_messages_for_value = [{"role": "assistant", "content":
|
134 |
"Hello! I'm an AI assistant. I'm currently running in a CPU-only "
|
135 |
"environment for efficient demonstration. How can I help you today?"
|
136 |
}]
|
137 |
|
138 |
+
# Define the Chatbot component without initial value in the constructor
|
139 |
+
chatbot_component = gr.Chatbot(height=500, type='messages')
|
140 |
+
|
141 |
demo = gr.ChatInterface(
|
142 |
fn=predict_chat,
|
143 |
+
# Pass the pre-defined chatbot_component
|
144 |
+
chatbot=chatbot_component,
|
145 |
textbox=gr.Textbox(
|
146 |
placeholder="Ask me a question...",
|
147 |
container=False,
|
|
|
164 |
clear_btn="Clear Chat"
|
165 |
)
|
166 |
|
167 |
+
# NEW LINE: Set the initial value after the demo (ChatInterface) is created
|
168 |
+
# This specifically targets the internal chatbot component that ChatInterface uses.
|
169 |
+
demo.chatbot.value = initial_messages_for_value
|
170 |
+
|
171 |
demo.launch()
|