Spaces:

anzorq
/

zedzek

Running on CPU Upgrade

anzorq commited on Apr 27, 2024

Commit

bf0e2b8

verified ·

1 Parent(s): 5308860

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -112,7 +112,7 @@ import transformers
 translator = ctranslate2.Translator("ctranslate2")
 tokenizer = transformers.AutoTokenizer.from_pretrained("anzorq/m2m100_418M_ft_ru-kbd_44K")
-@spaces.GPU
 def translate(text, num_beams=4, num_return_sequences=4):
     num_return_sequences = min(num_return_sequences, num_beams)

 translator = ctranslate2.Translator("ctranslate2")
 tokenizer = transformers.AutoTokenizer.from_pretrained("anzorq/m2m100_418M_ft_ru-kbd_44K")
+@spaces.GPU(duration=1)
 def translate(text, num_beams=4, num_return_sequences=4):
     num_return_sequences = min(num_return_sequences, num_beams)