Update README.md

README.md

@@ -1,35 +1,16 @@
 ---
 tags:
 - image-classification
+license: mit
 ---
-# Uno Card Detector AI Model
+## Uno Card Detector AI
 
-This repository contains the **Uno Card Detector AI Model** designed to identify Uno cards from images using a custom-trained neural network.
+This is a model for detecting Uno cards from images. You can try out the model directly by using the Hugging Face Inference API below.
 
-## Model Description
+[![Use Model](https://img.shields.io/badge/Use%20Model-HuggingFace-blue)](https://huggingface.co/szili2011/uno-card-detector-ai)
 
-This model was trained to detect and classify Uno cards using a dataset of card images. It uses convolutional neural networks (CNNs) to process the images and output predictions for the card type (e.g., colors, special cards). The model is capable of identifying cards in real-time, making it suitable for card-based games or recognition systems.
+### Model Details
 
-### Files Included:
-- **`pytorch_model.bin`**: The model's trained weights.
-- **`config.json`**: Configuration file for the model architecture and hyperparameters.
-- **`metadata.json`**: Metadata including information about the dataset and training process.
-
-## How to Use the Model
-
-### Load the Model in Python:
-
-To load and use the model, you can use the Hugging Face `transformers` or `huggingface_hub` library. Here's an example using `huggingface_hub`:
-
-```python
-from huggingface_hub import hf_hub_download
-
-# Download the model weights and configuration
-model_file = hf_hub_download(repo_id="szili2011/uno-card-detector-ai", filename="pytorch_model.bin")
-config_file = hf_hub_download(repo_id="szili2011/uno-card-detector-ai", filename="config.json")
-
-## Try the Model
-
-You can use the Uno Card Detector AI model via Hugging Face's Inference API:
-
-[![Use Model](https://img.shields.io/badge/Use%20Model-HuggingFace-blue)](https://huggingface.co/szili2011/uno-card-detector-new)
+- Task: Image Classification (Uno Cards)
+- Framework: PyTorch
+- Dataset: Custom Uno card dataset