Update app.py

app.py

@@ -114,4 +114,38 @@ 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"):
+        chat_interface = gr.Interface(
+            fn=generate_response,
+            inputs="text",
+            outputs="text",
+            title="Chat avec le modèle",
+            description="Entrez votre message pour obtenir une réponse du modèle"
+        )
+
+    with gr.Tab("Fine-Tuning"):
+        with gr.Row():
+            dataset_path = gr.Textbox(label="Chemin du dataset")
+            dataset_file = gr.Textbox(label="Nom du fichier du dataset")
+            epochs = gr.Number(label="Nombre d'époques", value=1)
+            batch_size = gr.Number(label="Taille du batch", value=2)
+            prefix = gr.Textbox(label="Préfixe pour les fichiers sauvegardés")
+
+            fine_tune_button = gr.Button("Lancer le Fine-Tuning")
+
+        fine_tune_output = gr.Textbox(label="État du Fine-Tuning")
+
+        fine_tune_button.click(
+            fine_tune_model,
+            inputs=[dataset_path, dataset_file, epochs, batch_size, prefix],
+            outputs=fine_tune_output
+        )
+
+        
+# Lancement de la démo
+if __name__ == "__main__":
+    demo.launch()
+