import gradio as gr
from transformers import pipeline
from PIL import Image
import torch

# Load your model
device = 0 if torch.cuda.is_available() else -1
pipe = pipeline("image-classification", model="beingamit99/car_damage_detection", device=device)

def predict_damage(image):
if image.mode != "RGB":
image = image.convert("RGB")
results = pipe(image)
return results

# Create the Gradio interface
iface = gr.Interface(
fn=predict_damage,
inputs=gr.Image(type="pil"),
outputs=gr.JSON(),
title="Car Damage Detection API",
description="Upload an image of a car to detect damages."
)

if __name__ == "__main__":
iface.launch()