Update app.py
Browse files
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() #
|
| 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 |
)
|