Update app.py

app.py

@@ -6,15 +6,16 @@ import torch
 #                 model="Salesforce/blip-image-captioning-base")
 
 pipe = pipeline(task="translation",
-                      model="./models/facebook/nllb-200-distilled-600M",
+                      model="facebook/nllb-200-distilled-600M",
                       torch_dtype=torch.bfloat16) 
 
-text_translated = pipe(text,
-                             src_lang="eng_Latn",
-                             tgt_lang="fra_Latn")
+# text_translated = pipe(text,
+#                              src_lang="eng_Latn",
+#                              tgt_lang="fra_Latn")
 
 def launch(input):
-    out = pipe(input)
+    out = pipe(input, src_lang="eng_Latn",
+                             tgt_lang="fra_Latn")
     return out[0]['generated_text']
 
 iface = gr.Interface(launch,