Update app.py

app.py

@@ -1,11 +1,11 @@
-
+import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 import torch
 
 # this model was loaded from https://hf.co/models
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
-model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-1.3B").to(device)
-tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-1.3B")
+model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M").to(device)
+tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M")
 
 
 LANGS = ["pes_Arab", "ckb_Arab", "eng_Latn"]