Moore-Language-Space-ZeroGPU

Sleeping

ArissBandoss commited on May 19

Commit

488bb16

verified ·

1 Parent(s): fb3e214

Update goai_helpers/goai_traduction.py

Files changed (1) hide show

goai_helpers/goai_traduction.py CHANGED Viewed

@@ -12,7 +12,7 @@ login(token=auth_token)
 @spaces.GPU
-def goai_traduction_debug(text, src_lang, tgt_lang):
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     if src_lang == "fra_Latn" and tgt_lang == "mos_Latn":

 @spaces.GPU
+def goai_traduction(text, src_lang, tgt_lang):
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     if src_lang == "fra_Latn" and tgt_lang == "mos_Latn":