Update app.py

app.py

@@ -1,10 +1,29 @@
 import os
 from huggingface_hub import login
+from datasets import load_dataset
+import torch
+from ultralytics import YOLO
 
 # Token'ı ortam değişkeninden alıyoruz
 token = os.getenv("HUGGINGFACE_TOKEN")
 login(token=token)
 
-from datasets import load_dataset
-
+# Veri setini yükle
 dataset = load_dataset("AtaGurbuz/Basic_Dataset")
+
+# Modeli eğitme işlemini başlatacağız
+def train_model():
+    # YOLOv8 modelini indiriyoruz (YOLOv8'i Hugging Face'de kullanacağız)
+    model = YOLO('yolov8n.pt')  # Küçük model, büyük modeli indirmek isterseniz yolov8l.pt kullanabilirsiniz.
+
+    # YOLO formatındaki veriyi kullanarak eğitimi başlatıyoruz
+    model.train(
+        data='path_to_yolo_data.yaml',  # YOLO formatındaki veri setinin konfigürasyon dosyası
+        epochs=100,  # Eğitim epoch sayısı
+        imgsz=640,  # Resim boyutu
+        batch=16,  # Batch boyutu
+        name='yolo_training'  # Eğitim sonucu kayıt ismi
+    )
+
+if __name__ == "__main__":
+    train_model()