cuda

app.py

@@ -218,7 +218,7 @@ tokenize_morph_simplified = lambda t: tokenize(morph_simplified_tokenizer, t)
 tokenize_case = lambda t: tokenize(case_tokenizer, t)
 
 def translate(model, tokenizer, text):
-  translated = model.generate(**tokenizer(text, return_tensors="pt", padding=True, truncation=True).to('cuda'))
+  translated = model.generate(**tokenizer(text, return_tensors="pt", padding=True, truncation=True))
   translated_line = [tokenizer.decode(t, skip_special_tokens=True) for t in translated]
   return translated_line