Update app.py

app.py

@@ -51,12 +51,21 @@ def generate_response(input_text):
 
 
 def fine_tune_model(dataset_path, dataset_file, epochs, batch_size, prefix):
+    import json  # Assurez-vous que json est importé
+
     # Récupération du fichier à partir de l'URL fournie
     response = requests.get(dataset_path)
     dataset_lines = response.text.strip().split('\n')
 
-    # Convertir les lignes en dictionnaires
-    dataset_dict = [json.loads(line) for line in dataset_lines]
+    # Vérifier si le fichier est au format JSONL ou JSON
+    if dataset_path.endswith('.jsonl'):
+        # Convertir les lignes en dictionnaires pour JSONL
+        dataset_dict = [json.loads(line) for line in dataset_lines if line]
+    elif dataset_path.endswith('.json'):
+        # Charger le fichier JSON directement
+        dataset_dict = json.loads(response.text)
+    else:
+        raise ValueError("Format de fichier non supporté. Utilisez .jsonl ou .json.")
 
     # Créer un Dataset Hugging Face
     dataset = Dataset.from_dict({
@@ -114,6 +123,7 @@ def fine_tune_model(dataset_path, dataset_file, epochs, batch_size, prefix):
     return "Fine-tuning terminé et modèle sauvegardé."
 
 
+
 # Interface Gradio
 with gr.Blocks() as demo:
     with gr.Tab("Chatbot"):