V1GPT2

Chat gpt2/chat2.py

@@ -0,0 +1,26 @@
+import gradio as gr
+from transformers import pipeline, set_seed
+
+# Initialiser le générateur de texte avec GPT-2
+generator = pipeline('text-generation', model='gpt2')
+
+# Optionnel : Fixer une graine aléatoire pour la reproductibilité
+set_seed(42)
+
+# Fonction de génération de texte
+def generate_text(prompt):
+    response = generator(prompt, max_length=100, num_return_sequences=1, truncation=True)
+    return response[0]['generated_text']
+
+# Définir une fonction pour l'interface de chat
+def chatbot(message,history):
+    str(message)
+    return generate_text(message)
+    
+
+
+
+gr.ChatInterface(
+    chatbot,
+    stop_btn="stop"
+    ).launch()