Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
4a3718d
1
Parent(s):
7d6b74d
Refactor predict function to retrieve user message using get_user_message method
Browse files
app.py
CHANGED
|
@@ -231,7 +231,7 @@ def predict(state,
|
|
| 231 |
|
| 232 |
history = state.get_history()
|
| 233 |
logger.info(f"==== History ====\n{history}")
|
| 234 |
-
message =
|
| 235 |
|
| 236 |
logger.info(f"==== Lenght Pixel values ====\n{len(pixel_values)}")
|
| 237 |
|
|
@@ -294,6 +294,7 @@ def ai_bot(
|
|
| 294 |
# Stream output
|
| 295 |
logger.info(f"==== Image paths ====\n{all_image_paths}")
|
| 296 |
|
|
|
|
| 297 |
response, _ = predict(state,
|
| 298 |
all_image_paths[0] if len(all_image_paths) > 0 else None,
|
| 299 |
max_input_tiles,
|
|
|
|
| 231 |
|
| 232 |
history = state.get_history()
|
| 233 |
logger.info(f"==== History ====\n{history}")
|
| 234 |
+
_, message = state.get_user_message(source=state.USER, position='last')
|
| 235 |
|
| 236 |
logger.info(f"==== Lenght Pixel values ====\n{len(pixel_values)}")
|
| 237 |
|
|
|
|
| 294 |
# Stream output
|
| 295 |
logger.info(f"==== Image paths ====\n{all_image_paths}")
|
| 296 |
|
| 297 |
+
|
| 298 |
response, _ = predict(state,
|
| 299 |
all_image_paths[0] if len(all_image_paths) > 0 else None,
|
| 300 |
max_input_tiles,
|