Spaces:
Runtime error
Runtime error
Update tabs/batch_processing.py
Browse files- tabs/batch_processing.py +4 -3
tabs/batch_processing.py
CHANGED
|
@@ -4,11 +4,12 @@ from utils.predictions import predict_batch_visual
|
|
| 4 |
def create_batch_processing_tab(model, class_names, device):
|
| 5 |
"""Create the batch processing tab"""
|
| 6 |
|
| 7 |
-
def batch_predict_wrapper(files
|
| 8 |
if not files:
|
| 9 |
return None, "❌ No files uploaded. Please select images to process."
|
| 10 |
|
| 11 |
-
|
|
|
|
| 12 |
return gallery, results
|
| 13 |
|
| 14 |
gr.Markdown("""
|
|
@@ -21,7 +22,7 @@ def create_batch_processing_tab(model, class_names, device):
|
|
| 21 |
batch_input = gr.File(
|
| 22 |
file_count="multiple",
|
| 23 |
label="Upload Multiple Images",
|
| 24 |
-
|
| 25 |
)
|
| 26 |
batch_btn = gr.Button("🚀 Process All Images", variant="primary", size="lg")
|
| 27 |
|
|
|
|
| 4 |
def create_batch_processing_tab(model, class_names, device):
|
| 5 |
"""Create the batch processing tab"""
|
| 6 |
|
| 7 |
+
def batch_predict_wrapper(files):
|
| 8 |
if not files:
|
| 9 |
return None, "❌ No files uploaded. Please select images to process."
|
| 10 |
|
| 11 |
+
# No progress parameter in Gradio 6
|
| 12 |
+
gallery, results = predict_batch_visual(files, model, class_names, device)
|
| 13 |
return gallery, results
|
| 14 |
|
| 15 |
gr.Markdown("""
|
|
|
|
| 22 |
batch_input = gr.File(
|
| 23 |
file_count="multiple",
|
| 24 |
label="Upload Multiple Images",
|
| 25 |
+
type="filepath"
|
| 26 |
)
|
| 27 |
batch_btn = gr.Button("🚀 Process All Images", variant="primary", size="lg")
|
| 28 |
|