tech5's picture
Upload 3 files
8f5eb48 verified
raw
history blame
907 Bytes
import os
# Create the app folder if it doesn't exist
os.makedirs('/content/BCCD-object/app', exist_ok=True)
# Define the Gradio app code
app_code = """
import gradio as gr
from ultralytics import YOLO
from PIL import Image
# Load the trained model
model = YOLO('/content/BCCD-object/model/bccd_yolov8.pt')
# Prediction function
def predict(image):
results = model(image)
annotated_image = results[0].plot() # Draw bounding boxes
return Image.fromarray(annotated_image)
# Gradio Interface
iface = gr.Interface(
fn=predict,
inputs=gr.Image(type="pil"),
outputs=gr.Image(type="pil"),
title="BCCD Object Detection",
description="Upload an image to detect RBC, WBC, and Platelets.",
)
# Launch the app
iface.launch(share=True)
"""
# Save the code to app.py
with open('/content/BCCD-object/app/app.py', 'w') as f:
f.write(app_code)