katielink commited on
Commit
c4f4d6f
1 Parent(s): 42bc81f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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
- #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,7 +31,7 @@ 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,
 
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,