from transformers import pipeline
import gradio as gr

classifier = pipeline("image-classification", model="prithivMLmods/flood-image-detection")

def classify(image):
    results = classifier(image)
    return {r["label"]: r["score"] for r in results}

gr.Interface(
    fn=classify,
    inputs=gr.Image(type="pil"),
    outputs=gr.Label(),
    title="Flood Detector"
).launch()