Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
|
| 3 |
############### VANILLA INFERENCE ###############
|
|
@@ -111,6 +112,7 @@ import transformers
|
|
| 111 |
translator = ctranslate2.Translator("ctranslate2")
|
| 112 |
tokenizer = transformers.AutoTokenizer.from_pretrained("anzorq/m2m100_418M_ft_ru-kbd_44K")
|
| 113 |
|
|
|
|
| 114 |
def translate(text, num_beams=4, num_return_sequences=4):
|
| 115 |
|
| 116 |
num_return_sequences = min(num_return_sequences, num_beams)
|
|
|
|
| 1 |
+
import spaces
|
| 2 |
import gradio as gr
|
| 3 |
|
| 4 |
############### VANILLA INFERENCE ###############
|
|
|
|
| 112 |
translator = ctranslate2.Translator("ctranslate2")
|
| 113 |
tokenizer = transformers.AutoTokenizer.from_pretrained("anzorq/m2m100_418M_ft_ru-kbd_44K")
|
| 114 |
|
| 115 |
+
+@spaces.GPU
|
| 116 |
def translate(text, num_beams=4, num_return_sequences=4):
|
| 117 |
|
| 118 |
num_return_sequences = min(num_return_sequences, num_beams)
|