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() |