alex-abb commited on
Commit
2c2ec58
·
verified ·
1 Parent(s): 3f78b1e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,11 +8,11 @@ model = BlenderbotForConditionalGeneration.from_pretrained("facebook/blenderbot-
8
  # Initialiser le générateur de texte avec GPT-2
9
 
10
  # Optionnel : Fixer une graine aléatoire pour la reproductibilit
11
- @spaces.GPU()
12
  # Fonction de génération de texte
13
  def generate_text(prompt):
14
 
15
- inputs = tokenizer(text, return_tensors="pt")
16
 
17
  response_ids = model.generate(inputs.input_ids)
18
 
 
8
  # Initialiser le générateur de texte avec GPT-2
9
 
10
  # Optionnel : Fixer une graine aléatoire pour la reproductibilit
11
+ @spaces.GPU
12
  # Fonction de génération de texte
13
  def generate_text(prompt):
14
 
15
+ inputs = tokenizer(prompt, return_tensors="pt")
16
 
17
  response_ids = model.generate(inputs.input_ids)
18