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