Spaces:

lexi96
/

biogpt-demo

Runtime error

lexi96 commited on Mar 1, 2023

Commit

ddc467f

•

1 Parent(s): 45b5bf8

up

Files changed (1) hide show

app.py CHANGED Viewed

@@ -23,7 +23,7 @@ model = AutoModelForCausalLM.from_pretrained("microsoft/BioGPT-Large-PubMedQA")
 pipe_biogpt = pipeline("text-generation",
                        model=model,
                        tokenizer = tokenizer,
-                       device="cuda:0")
 title = "BioGPT Q&A Demo"
 description = """

 pipe_biogpt = pipeline("text-generation",
                        model=model,
                        tokenizer = tokenizer,
+                       device="cpu:0")
 title = "BioGPT Q&A Demo"
 description = """