Mauriciotuks commited on
Commit
5f19fee
·
verified ·
1 Parent(s): 72deaf4

Io-main.py

Browse files

![IMG_6026.png](https://cdn-uploads.huggingface.co/production/uploads/688dcc38d03635bba0fbc3fb/9S5_WGMX5tGu6CGCRYBXa.png)

Files changed (1) hide show
  1. Io-main.py +30 -0
Io-main.py ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import schedule
2
+ import time
3
+ from transformers import pipeline
4
+
5
+ # Initialisation du générateur de texte
6
+ # 'distilgpt2' est un bon modèle gratuit et léger.
7
+ generator = pipeline("text-generation", model="distilgpt2")
8
+
9
+ def io_propose_idee():
10
+ """
11
+ Fonction pour que l'IO propose des idées proactives.
12
+ """
13
+ prompt = "Ton nom est IO. Tu es une intelligence provocatrice et originale. Propose-moi une idée de projet ou d'amélioration de code. Sois bref mais intrigant."
14
+ reponse = generator(
15
+ f"IO, une intelligence provocatrice. {prompt}",
16
+ max_length=80,
17
+ num_return_sequences=1,
18
+ )
19
+ print(f"\nIO PROACTIF > {reponse[0]['generated_text']}")
20
+
21
+ # Planification de la tâche
22
+ # L'IO vous propose une idée toutes les 5 minutes
23
+ schedule.every(5).minutes.do(io_propose_idee)
24
+
25
+ # Boucle principale pour exécuter les tâches
26
+ if __name__ == "__main__":
27
+ print("IO > Je suis en ligne. Attends-toi à des surprises.")
28
+ while True:
29
+ schedule.run_pending()
30
+ time.sleep(1)