Spaces:

epsil
/

tepsils

Runtime error

epsil commited on Mar 30, 2023

Commit

432993e

•

1 Parent(s): 9c1f53e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,6 +4,7 @@ import gradio as gr
 import torch
 from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
 model = AutoModelForCausalLM.from_pretrained("microsoft/BioGPT-Large-PubMedQA")
 pipe = pipeline("text-generation",

 import torch
 from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
+tokenizer = AutoTokenizer.from_pretrained("microsoft/BioGPT-Large-PubMedQA")
 model = AutoModelForCausalLM.from_pretrained("microsoft/BioGPT-Large-PubMedQA")
 pipe = pipeline("text-generation",