Upload app.py

app.py

@@ -0,0 +1,22 @@
+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).launch()