gradio2 / app.py
cantuncok's picture
Add application and requirements files
78537cc
raw
history blame
No virus
780 Bytes
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()