Update app.py

app.py

@@ -19,8 +19,10 @@ with st.form("Prediction_form"):
    submit = st.form_submit_button("Translate Text to Hindi")
    #
    if submit:
-        tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M",use_auth_token=True)
-        model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M",use_auth_token=True)
+        #tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M",use_auth_token=True)
+        tokenizer = AutoTokenizer.from_pretrained(".",use_auth_token=True)
+        #model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M",use_auth_token=True)
+        model = AutoModelForSeq2SeqLM.from_pretrained(".",use_auth_token=True)
         inputs = tokenizer(text, return_tensors="pt")
 
         translated_tokens = model.generate(**inputs,