Update app.py

app.py

@@ -36,11 +36,8 @@ def biogpt(
 ):
     
     en_prompt = translate_to_english(prompt, lang_model_id, base_lang)
-    model = BioGptForCausalLM.from_pretrained(biogpt_model_id)
-    tokenizer = BioGptTokenizer.from_pretrained(biogpt_model_id)
-    generator = pipeline('text-generation', model=model, tokenizer=tokenizer, device='cuda:0')
-    set_seed(42)
-    output = generator(en_prompt, max_length=max_length, num_return_sequences=num_return_sequences, do_sample=True)
+    generator = pipeline('text-generation', device='cuda:0')
+    output = generator(en_prompt, model=biogpt_model_id, 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']