Upload 7 files

Image Classification first commit

.gitattributes

@@ -33,3 +33,6 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+examples/car.jpg filter=lfs diff=lfs merge=lfs -text
+examples/cat.jpg filter=lfs diff=lfs merge=lfs -text
+examples/ship.jpg filter=lfs diff=lfs merge=lfs -text

app.py

@@ -0,0 +1,48 @@
+import gradio as gr
+from transformers import pipeline
+
+# Load models
+cifar10_classifier = pipeline("image-classification", model="Fadri/results")
+clip_detector = pipeline(model="openai/clip-vit-large-patch14", task="zero-shot-image-classification")
+
+# CIFAR-10 Klassen
+labels_cifar10 = [
+    'airplane', 'automobile', 'bird', 'cat', 'deer', 
+    'dog', 'frog', 'horse', 'ship', 'truck'
+]
+
+def classify_image(image):
+    # Klassifikation mit deinem trainierten CIFAR-10 Modell
+    cifar10_results = cifar10_classifier(image)
+    cifar10_output = {result['label']: result['score'] for result in cifar10_results}
+    
+    # Zero-Shot-Klassifikation mit CLIP
+    clip_results = clip_detector(image, candidate_labels=labels_cifar10)
+    clip_output = {result['label']: result['score'] for result in clip_results}
+    
+    return {
+        "CIFAR-10 ViT Klassifikation": cifar10_output, 
+        "CLIP Zero-Shot Klassifikation": clip_output
+    }
+
+# Beispielbilder - Du kannst diese später anpassen
+example_images = [
+    ["examples/airplane.jpg"],
+    ["examples/car.jpg"],
+    ["examples/dog.jpg"],
+    ["examples/cat.jpg"],
+    ["examples/ship.jpg"],
+    ["examples/truck.jpg"]
+]
+
+# Gradio Interface
+iface = gr.Interface(
+    fn=classify_image,
+    inputs=gr.Image(type="filepath"),
+    outputs=gr.JSON(),
+    title="CIFAR-10 Klassifikation",
+    description="Lade ein Bild hoch und vergleiche die Ergebnisse zwischen deinem trainierten ViT Modell und dem Zero-Shot CLIP Modell für CIFAR-10 Klassen.",
+    examples=example_images
+)
+
+iface.launch()

requirements.txt

@@ -0,0 +1,4 @@
+gradio
+transformers
+torch
+torchvision