Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -244,10 +244,13 @@ def process_video(video_file):
|
|
| 244 |
|
| 245 |
# Step 2: Generate embeddings (GPU operation)
|
| 246 |
status_msg += "\nπ Generating embeddings for video clips..."
|
| 247 |
-
|
| 248 |
|
| 249 |
-
|
| 250 |
-
|
|
|
|
|
|
|
|
|
|
| 251 |
|
| 252 |
status_msg += f"\nβ
Generated embeddings for {len(embeddings)} clips"
|
| 253 |
status_msg += "\nπ― Ready for queries!"
|
|
@@ -262,9 +265,9 @@ def search_video_clips(query, video_embeddings_state, video_clips_state):
|
|
| 262 |
|
| 263 |
# Create Gradio interface
|
| 264 |
def create_interface():
|
| 265 |
-
with gr.Blocks(title="Security Camera AI Assistant", theme=gr.themes.
|
| 266 |
gr.Markdown("# π₯ Security Camera AI Assistant")
|
| 267 |
-
gr.Markdown("Upload
|
| 268 |
|
| 269 |
with gr.Row():
|
| 270 |
with gr.Column(scale=1):
|
|
|
|
| 244 |
|
| 245 |
# Step 2: Generate embeddings (GPU operation)
|
| 246 |
status_msg += "\nπ Generating embeddings for video clips..."
|
| 247 |
+
embeddings_result = gpu_operation("generate_embeddings", clip_paths)
|
| 248 |
|
| 249 |
+
# Check if embeddings_result is an error tuple (string, list)
|
| 250 |
+
if isinstance(embeddings_result, tuple) and len(embeddings_result) == 2 and isinstance(embeddings_result[0], str) and "β" in embeddings_result[0]:
|
| 251 |
+
return embeddings_result[0], clips_info, [], [] # Return error message
|
| 252 |
+
|
| 253 |
+
embeddings = embeddings_result # Assign actual embeddings
|
| 254 |
|
| 255 |
status_msg += f"\nβ
Generated embeddings for {len(embeddings)} clips"
|
| 256 |
status_msg += "\nπ― Ready for queries!"
|
|
|
|
| 265 |
|
| 266 |
# Create Gradio interface
|
| 267 |
def create_interface():
|
| 268 |
+
with gr.Blocks(title="Security Camera AI Assistant", theme=gr.themes.Ocean()) as demo:
|
| 269 |
gr.Markdown("# π₯ Security Camera AI Assistant")
|
| 270 |
+
gr.Markdown("Upload a video.Then search sing natural language queries, to get relevant clips with precise timestamps!!")
|
| 271 |
|
| 272 |
with gr.Row():
|
| 273 |
with gr.Column(scale=1):
|