Update app.py

app.py

@@ -6,11 +6,12 @@ import spaces
 device = "cuda" 
 model_id = "leks-forever/nllb-200-distilled-600M-v1"
 
+tokenizer = NllbTokenizer.from_pretrained(model_id)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_id)
+
 @spaces.GPU
 def translate(text, src_lang='lez_Cyrl', tgt_lang='rus_Cyrl', a=32, b=3, max_input_length=1024, num_beams=1, **kwargs):
-    
-    tokenizer = NllbTokenizer.from_pretrained(model_id)
-    model = AutoModelForSeq2SeqLM.from_pretrained(model_id)
+
     model = model.to(device)
 
     if src_lang in language_codes: