Spaces:
Paused
Paused
Update chatbot.py
Browse files- chatbot.py +5 -5
chatbot.py
CHANGED
@@ -243,12 +243,12 @@ def model_inference( user_prompt, chat_history, web_search):
|
|
243 |
yield output
|
244 |
else:
|
245 |
message = user_prompt
|
246 |
-
if len(message
|
247 |
-
image = [message
|
248 |
-
elif len(message
|
249 |
-
image = [msg.path for msg in message
|
250 |
|
251 |
-
txt = message
|
252 |
|
253 |
video_extensions = ("avi", "mp4", "mov", "mkv", "flv", "wmv", "mjpeg", "wav", "gif", "webm", "m4v", "3gp")
|
254 |
image_extensions = Image.registered_extensions()
|
|
|
243 |
yield output
|
244 |
else:
|
245 |
message = user_prompt
|
246 |
+
if len(message["files"]) == 1:
|
247 |
+
image = [message["files"][0].path]
|
248 |
+
elif len(message["files"]) > 1:
|
249 |
+
image = [msg.path for msg in message["files"]]
|
250 |
|
251 |
+
txt = message["text"]
|
252 |
|
253 |
video_extensions = ("avi", "mp4", "mov", "mkv", "flv", "wmv", "mjpeg", "wav", "gif", "webm", "m4v", "3gp")
|
254 |
image_extensions = Image.registered_extensions()
|