fatimaxa commited on
Commit
6fcb290
·
verified ·
1 Parent(s): 05c4e3d

Update tabs/batch_processing.py

Browse files
Files changed (1) hide show
  1. 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, progress=gr.Progress()):
8
  if not files:
9
  return None, "❌ No files uploaded. Please select images to process."
10
 
11
- gallery, results = predict_batch_visual(files, model, class_names, device, progress)
 
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
- file_types=["image"]
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