Spaces:

runaksh
/

Medquad_qna_model

Sleeping

runaksh commited on Sep 5, 2023

Commit

6ffbd3f

1 Parent(s): a378f63

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,8 +12,8 @@ from tensorflow import keras
 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):

 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):