Update app.py

app.py

@@ -36,7 +36,7 @@ def biogpt(
 ):
     
     en_prompt = translate_to_english(prompt, lang_model_id, base_lang)
-    generator = pipeline('text-generation', device='cuda:0')
+    generator = pipeline("text-generation", model=biogpt_model_id, 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):