Marcel0123 commited on
Commit
cec0dd2
·
verified ·
1 Parent(s): 7c0e884

Upload 3 files

Browse files
Files changed (3) hide show
  1. README.md +13 -13
  2. app.py +28 -0
  3. requirements.txt +4 -0
README.md CHANGED
@@ -1,13 +1,13 @@
1
- ---
2
- title: Astro Space
3
- emoji: 🏆
4
- colorFrom: pink
5
- colorTo: green
6
- sdk: gradio
7
- sdk_version: 5.38.2
8
- app_file: app.py
9
- pinned: false
10
- license: mit
11
- ---
12
-
13
- Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
1
+ # 🔭 AstroClassifier
2
+
3
+ **Wat is dit?**
4
+ Een AI-toepassing waarmee je afbeeldingen van ruimteobjecten kunt uploaden (planeten, sterren, nevels, zwarte gaten). De AI (Vision Transformer) probeert te voorspellen wat je ziet.
5
+
6
+ **Gebruikt model:**
7
+ `google/vit-base-patch16-224`
8
+
9
+ **Gemaakt met:**
10
+ Gradio + Hugging Face Spaces
11
+
12
+ **Let op:**
13
+ Het model is niet specifiek getraind op ruimteobjecten, dus de voorspellingen kunnen verrassend of humoristisch zijn. Educatief prototype!
app.py ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import ViTFeatureExtractor, ViTForImageClassification
2
+ from PIL import Image
3
+ import torch
4
+ import gradio as gr
5
+
6
+ # Model en extractor laden
7
+ model = ViTForImageClassification.from_pretrained("google/vit-base-patch16-224")
8
+ extractor = ViTFeatureExtractor.from_pretrained("google/vit-base-patch16-224")
9
+
10
+ # Classificatiefunctie
11
+ def classify_space_image(image):
12
+ inputs = extractor(images=image, return_tensors="pt")
13
+ with torch.no_grad():
14
+ logits = model(**inputs).logits
15
+ predicted = logits.argmax(-1).item()
16
+ label = model.config.id2label[predicted]
17
+ return f"Voorspelling: {label}"
18
+
19
+ # Gradio Interface
20
+ demo = gr.Interface(
21
+ fn=classify_space_image,
22
+ inputs=gr.Image(type="pil"),
23
+ outputs="text",
24
+ title="🔭 AstroClassifier",
25
+ description="Upload een ruimte-afbeelding (bijv. van NASA of een telescoop) en ontdek of het een ster, planeet, nevel of zwart gat is (volgens de AI!)."
26
+ )
27
+
28
+ demo.launch()
requirements.txt ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ transformers
2
+ torch
3
+ gradio
4
+ Pillow