Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -9,6 +9,16 @@ from trimesh.exchange.gltf import export_glb
|
|
| 9 |
import numpy as np
|
| 10 |
import tempfile
|
| 11 |
import copy
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
def predict(_chatbot, task_history):
|
| 13 |
chat_query = _chatbot[-1][0]
|
| 14 |
query = task_history[-1][0]
|
|
|
|
| 9 |
import numpy as np
|
| 10 |
import tempfile
|
| 11 |
import copy
|
| 12 |
+
|
| 13 |
+
def _remove_image_special(text):
|
| 14 |
+
text = text.replace('<ref>', '').replace('</ref>', '')
|
| 15 |
+
return re.sub(r'<box>.*?(</box>|$)', '', text)
|
| 16 |
+
|
| 17 |
+
|
| 18 |
+
def is_video_file(filename):
|
| 19 |
+
video_extensions = ['.mp4', '.avi', '.mkv', '.mov', '.wmv', '.flv', '.webm', '.mpeg']
|
| 20 |
+
return any(filename.lower().endswith(ext) for ext in video_extensions)
|
| 21 |
+
|
| 22 |
def predict(_chatbot, task_history):
|
| 23 |
chat_query = _chatbot[-1][0]
|
| 24 |
query = task_history[-1][0]
|