Spaces:
Runtime error
Runtime error
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,
|