Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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(
|
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 |
|