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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -13,22 +13,25 @@ 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 [output_biogpt, output_biomedlm]
 
 
 
25
 
26
  io = gr.Interface(
27
  inference,
28
  gr.Textbox(lines=3),
29
  outputs=[
30
  gr.Textbox(lines=3, label="BioGPT"),
31
- gr.Textbox(lines=3, label="BioMedLM")
32
  ],
33
  title=title,
34
  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(
30
  inference,
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,