update

app.py

@@ -39,22 +39,23 @@ def biogpt(
     generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
     set_seed(42)
     output = generator(en_prompt, max_length=max_length, num_return_sequences=num_return_sequences, do_sample=True)
-    
     output_dict = {}
     for i in range(num_return_sequences):
         output_dict[str(i+1)] = output[i]['generated_text']
-    
+
     output_text = ""
     for i in range(num_return_sequences):
         output_text += f'{output_dict[str(i+1)]}\n\n'
-
+    
     if base_lang == "English":
         base_lang_output = output_text
         
     else: 
-        base_lang_output = translate(lang_model_id, output_text, "en", lang_ids[base_lang])[0]
+        base_lang_output_ = ""
         for i in range(num_return_sequences):
-            base_lang_output += f'{output_dict[str(i+1)]}\n\n'
+            base_lang_output_ += f'{translate(lang_model_id, output_dict[str(i+1)], "en", lang_ids[base_lang])[0]}\n\n'
+        base_lang_output = base_lang_output_
+    
     
     return en_prompt, output_text, base_lang_output