Hugging Face's logo

Spaces:
alex-abb
/
8B-on-GPU-ZERO
Sleeping

App Files Community
alex-abb commited on
Commit
3f78b1e
·
verified ·
1 Parent(s): cefd640

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import gradio as gr
2
  from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration
3
-
4
 
5
  tokenizer = BlenderbotTokenizer.from_pretrained('facebook/blenderbot-8B')
6
  model = BlenderbotForConditionalGeneration.from_pretrained("facebook/blenderbot-8B")
@@ -8,7 +8,7 @@ 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
-
12
  # Fonction de génération de texte
13
  def generate_text(prompt):
14
 
 
1
  import gradio as gr
2
  from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration
3
+ import spaces
4
 
5
  tokenizer = BlenderbotTokenizer.from_pretrained('facebook/blenderbot-8B')
6
  model = BlenderbotForConditionalGeneration.from_pretrained("facebook/blenderbot-8B")
 
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