Update app.py

app.py

@@ -1,48 +1,49 @@
+import logging
 import os
-import requests
-import json
 from huggingface_hub import login
+from datasets import load_dataset
+import torch
 from ultralytics import YOLO
 
+# Logging yapılandırması
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+
 # Token'ı ortam değişkeninden alıyoruz
 token = os.getenv("HUGGINGFACE_TOKEN")
-login(token=token)
-
-# Hugging Face API URL'leri
-train_url = "https://datasets-server.huggingface.co/rows?dataset=AtaGurbuz%2FBasic_Dataset&split=train"
-val_url = "https://datasets-server.huggingface.co/rows?dataset=AtaGurbuz%2FBasic_Dataset&split=val"
-
-headers = {
-    "Authorization": f"Bearer {token}"
-}
-
-def download_data(url, split_name):
-    response = requests.get(url, headers=headers)
-    if response.status_code == 200:
-        data = response.json()
-        with open(f"{split_name}.json", "w") as f:
-            json.dump(data, f)
-        print(f"{split_name} verisi başarıyla indirildi.")
-    else:
-        print(f"{split_name} verisi indirilemedi. Hata: {response.status_code}")
-
-# Eğitim ve doğrulama verilerini indir
-download_data(train_url, "train")
-download_data(val_url, "val")
+if token:
+    logging.info("Hugging Face token bulundu.")
+    login(token=token)
+else:
+    logging.error("Hugging Face token bulunamadı. Lütfen token'ı doğru şekilde ayarlayın.")
+
+# Veri setini yükle
+try:
+    dataset = load_dataset("AtaGurbuz/Basic_Dataset")
+    logging.info("Veri seti başarıyla yüklendi.")
+except Exception as e:
+    logging.error(f"Veri seti yüklenemedi: {e}")
 
 # 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='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
-    )
+    try:
+        logging.info("YOLOv8 modeli indiriliyor...")
+        model = YOLO('yolov8n.pt')  # Küçük model, büyük modeli indirmek isterseniz yolov8l.pt kullanabilirsiniz.
+        logging.info("YOLOv8 modeli başarıyla indirildi.")
+
+        # YOLO formatındaki veriyi kullanarak eğitimi başlatıyoruz
+        logging.info("Model eğitimi başlatılıyor...")
+        model.train(
+            data='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
+        )
+        logging.info("Eğitim tamamlandı.")
+    except Exception as e:
+        logging.error(f"Eğitim sırasında bir hata oluştu: {e}")
 
 if __name__ == "__main__":
+    logging.info("Model eğitim süreci başlıyor...")
     train_model()
+    logging.info("Model eğitim süreci tamamlandı.")