from transformers import pipeline
import gradio as gr

# Modeli yükleyin
model_name = "cantuncok/autotrain1-model"
classifier = pipeline("image-classification", model=model_name)

# Görsel sınıflandırma fonksiyonu
def classify_image(image):
    # Modelden tahminleri alın
    results = classifier(image)
    # Tüm tahminleri ve puanlarını döndürün
    return {result['label']: result['score'] for result in results}

# Gradio arayüzünü oluşturun
interface = gr.Interface(
    fn=classify_image,
    inputs=gr.inputs.Image(type="pil"),
    outputs="label",
    title="Görüntü Sınıflandırma",
    description="Bu uygulama bir görüntüyü sınıflandırır ve etiketi döndürür."
)

# Uygulamayı başlatın
if __name__ == "__main__":
    interface.launch()