import gradio as gr
import torch

# Laad je model
model_path = "Tsiemonsma/app-archipelagos"  # Zorg ervoor dat dit pad klopt
model = torch.hub.load('Tsiemonsma/app-archipelagos', 'custom', path=model_path, force_reload=True)

def predict(image):
    results = model(image)  # Voer de detectie uit
    return results.pandas().xyxy[0]  # Retourneer de resultaten als een DataFrame

iface = gr.Interface(fn=predict, inputs="image", outputs="dataframe")
iface.launch()