Update app.py

app.py

@@ -7,7 +7,7 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 
 def load_models():
     # build model and tokenizer
-    model_name_dict = {'nllb-distilled-600M': 'facebook/nllb-200-distilled-600M',
+    model_name_dict = {'nllb-distilled-600M': 'facebook/nllb-3.3B',
                   #'nllb-1.3B': 'facebook/nllb-200-1.3B',
                   #'nllb-distilled-1.3B': 'facebook/nllb-200-distilled-1.3B',
                   #'nllb-3.3B': 'facebook/nllb-200-3.3B',
@@ -40,11 +40,7 @@ def translation(source, target, text):
     end_time = time.time()
 
     output = output[0]['translation_text']
-    result = {'inference_time': end_time - start_time,
-              'source': source,
-              'target': target,
-              'result': output}
-    return result
+    return output
 
 
 if __name__ == '__main__':