Update app.py

app.py

@@ -1,11 +1,18 @@
-from transformers import pipeline, set_seed
-from transformers import BioGptTokenizer, BioGptForCausalLM
 import gradio as gr
-model = BioGptForCausalLM.from_pretrained("microsoft/biogpt")
-tokenizer = BioGptTokenizer.from_pretrained("microsoft/biogpt")
-generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
-set_seed(42)
-def predict(text):
-    return generator(text, max_length=20, num_return_sequences=1, do_sample=True)
-interface = gr.Interface(predict, "textbox", "textbox", title="Biogpt", description="Biogpt")
-interface.launch()
+from transformers import pipeline
+
+generator = pipeline("text-generation", model="microsoft/BioGPT-Large")
+
+def generate_text(prompt):
+    result = generator(prompt, max_length=200, do_sample=True)
+    return result[0]['generated_text']
+
+demo = gr.Interface(
+    fn=generate_text,
+    inputs="text",
+    outputs="text",
+    title="BioGPT Text Generator",
+    description="Masukkan ayat bioperubatan dan hasilkan sambungan teks.",
+)
+
+demo.launch()