Didier commited on
Commit
6290a8c
·
verified ·
1 Parent(s): c551723

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -6,6 +6,7 @@ Description: Module to translate between 400 languages.
6
  Author: Didier Guillevic
7
  Date: 2024-09-07
8
  """
 
9
  import torch
10
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
11
  import gradio as gr
@@ -22,6 +23,7 @@ model = torch.compile(model)
22
  #
23
  # Translate given input text in given target language
24
  #
 
25
  def translate_text(
26
  text: str,
27
  tgt_lang: str,
 
6
  Author: Didier Guillevic
7
  Date: 2024-09-07
8
  """
9
+ import spaces
10
  import torch
11
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
12
  import gradio as gr
 
23
  #
24
  # Translate given input text in given target language
25
  #
26
+ @spaces.GPU
27
  def translate_text(
28
  text: str,
29
  tgt_lang: str,