Akheela commited on
Commit
2bdc739
1 Parent(s): 1797dc1

Upload exercice 8.py

Browse files
Files changed (1) hide show
  1. exercice 8.py +29 -0
exercice 8.py ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import GPT2LMHeadModel, GPT2Tokenizer
2
+
3
+ # Chargement du modèle GPT-2 et le tokenizer
4
+ model_name = "gpt2"
5
+ tokenizer = GPT2Tokenizer.from_pretrained(model_name)
6
+ model = GPT2LMHeadModel.from_pretrained(model_name)
7
+
8
+ #Le prompt
9
+ prompt = "Au début du 21ème siècle, les humains ont découvert une nouvelle technologie"
10
+
11
+ # Encodage du prompt pour l'entrée dans le modèle
12
+ inputs = tokenizer.encode(prompt, return_tensors='pt')
13
+
14
+ # Générer du texte par exemple 100 tokens
15
+ output = model.generate(
16
+ inputs,
17
+ max_length=100,
18
+ num_return_sequences=1,
19
+ do_sample = True,
20
+ temperature = 0.7,
21
+ top_k = 50,
22
+ top_p = 0.95
23
+ )
24
+
25
+ # Décodage de la génération en texte compréhensible
26
+ generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
27
+
28
+ #Affichage
29
+ print(generated_text)