nimalan commited on
Commit
b9d7aff
·
verified ·
1 Parent(s): 55addc4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -21,7 +21,7 @@ models = {
21
  def run_inference(model_name, image: Image.Image):
22
  model = models[model_name]
23
  results = model.predict(source=np.array(image), imgsz=640, conf=0.25, verbose=False)
24
- annotated = results[0].plot() # Annotated image with boxes and labels
25
  return Image.fromarray(annotated)
26
 
27
  demo = gr.Interface(
@@ -30,7 +30,7 @@ demo = gr.Interface(
30
  gr.Dropdown(choices=list(models.keys()), label="Select YOLO Model"),
31
  gr.Image(type="pil", label="Upload Image"),
32
  ],
33
- outputs=gr.Image(label="Detected Output"),
34
  title="MH WEED16 YOLO Inference",
35
  description="Upload an image to detect weed species using a selected YOLO model.",
36
  )
 
21
  def run_inference(model_name, image: Image.Image):
22
  model = models[model_name]
23
  results = model.predict(source=np.array(image), imgsz=640, conf=0.25, verbose=False)
24
+ annotated = results[0].plot() # Returns a numpy array (RGB)
25
  return Image.fromarray(annotated)
26
 
27
  demo = gr.Interface(
 
30
  gr.Dropdown(choices=list(models.keys()), label="Select YOLO Model"),
31
  gr.Image(type="pil", label="Upload Image"),
32
  ],
33
+ outputs=gr.Image(type="filepath", label="Detected Output"), # ✅ required for GitHub integration
34
  title="MH WEED16 YOLO Inference",
35
  description="Upload an image to detect weed species using a selected YOLO model.",
36
  )