Updated Dockerfile for Hugging Face Spaces deployment

models/nllb.py

@@ -27,9 +27,7 @@ def nllb():
     # Load models
     tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-1.3B")
     model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-1.3B").to(device)
-    # write done to the file named status.txt
-    with open("status.txt", 'w') as f:
-        f.write("done")
+    
     return model, tokenizer
 
 def nllb_translate(model, tokenizer, article, language):

status.txt

@@ -1 +0,0 @@
-done