Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -13,17 +13,17 @@ examples = [['COVID-19 is'],['A 65-year-old female patient with a past medical h
|
|
13 |
#print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
|
14 |
|
15 |
pipe_biogpt = pipeline("text-generation", model="microsoft/biogpt")
|
16 |
-
|
17 |
|
18 |
title = "Compare generative biomedical LLMs!"
|
19 |
description = "This demo compares [BioGPT](https://huggingface.co/microsoft/biogpt) and [BioMedLM](https://huggingface.co/stanford-crfm/BioMedLM)."
|
20 |
|
21 |
def inference(text):
|
22 |
output_biogpt = pipe_biogpt(text, max_length=100)[0]["generated_text"]
|
23 |
-
|
24 |
return [
|
25 |
output_biogpt,
|
26 |
-
|
27 |
]
|
28 |
|
29 |
io = gr.Interface(
|
@@ -31,7 +31,7 @@ io = gr.Interface(
|
|
31 |
gr.Textbox(lines=3),
|
32 |
outputs=[
|
33 |
gr.Textbox(lines=3, label="BioGPT"),
|
34 |
-
|
35 |
],
|
36 |
title=title,
|
37 |
description=description,
|
|
|
13 |
#print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
|
14 |
|
15 |
pipe_biogpt = pipeline("text-generation", model="microsoft/biogpt")
|
16 |
+
pipe_biomedlm = pipeline("text-generation", model="stanford-crfm/BioMedLM")
|
17 |
|
18 |
title = "Compare generative biomedical LLMs!"
|
19 |
description = "This demo compares [BioGPT](https://huggingface.co/microsoft/biogpt) and [BioMedLM](https://huggingface.co/stanford-crfm/BioMedLM)."
|
20 |
|
21 |
def inference(text):
|
22 |
output_biogpt = pipe_biogpt(text, max_length=100)[0]["generated_text"]
|
23 |
+
output_biomedlm = pipe_biomedlm(text, max_length=100)[0]["generated_text"]
|
24 |
return [
|
25 |
output_biogpt,
|
26 |
+
output_biomedlm
|
27 |
]
|
28 |
|
29 |
io = gr.Interface(
|
|
|
31 |
gr.Textbox(lines=3),
|
32 |
outputs=[
|
33 |
gr.Textbox(lines=3, label="BioGPT"),
|
34 |
+
gr.Textbox(lines=3, label="BioMedLM")
|
35 |
],
|
36 |
title=title,
|
37 |
description=description,
|