Spaces:

runaksh
/

Medquad_qna_model

Sleeping

runaksh commited on Sep 5, 2023

Commit

f06cd88

•

1 Parent(s): 675fbf9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,8 +14,8 @@ from transformers import GPT2Tokenizer, GPT2LMHeadModel
 from transformers import AutoModelWithLMHead, AutoTokenizer
 import numpy as np
-loaded_model = GPT2LMHeadModel.from_pretrained("gpt2")
-loaded_tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
 def generate_query_response(prompt, max_length=200):

 from transformers import AutoModelWithLMHead, AutoTokenizer
 import numpy as np
+loaded_model = AutoModelWithLMHead.from_pretrained("runaksh/medquad-finetuned-gpt2")
+loaded_tokenizer = AutoTokenizer.from_pretrained("runaksh/medquad-finetuned-gpt2")
 def generate_query_response(prompt, max_length=200):