Spaces:

Shariar00
/

Medical_data_Fine_Tune_model_QnA

Runtime error

Shariar00 commited on Jan 28

Commit

ec75ddf

•

1 Parent(s): 1b6bc75

update tokenizer

Files changed (1) hide show

app.py CHANGED Viewed

@@ -21,7 +21,7 @@ def generate_summary(input_text):
     # Create a text generation pipeline
     # text_generation_pipeline = pipeline("Falconsai/medical_summarization", model=model, tokenizer=tokenizer)
-    tokenizer = AutoTokenizer.from_pretrained("Shariar00/medical_summarization_finetune_medical_qa")
     model = AutoModelForSeq2SeqLM.from_pretrained("Shariar00/medical_summarization_finetune_medical_qa")
     text_generation_pipeline = pipeline("summarization", model=model, tokenizer=tokenizer)

     # Create a text generation pipeline
     # text_generation_pipeline = pipeline("Falconsai/medical_summarization", model=model, tokenizer=tokenizer)
+    tokenizer = AutoTokenizer.from_pretrained("Falconsai/medical_summarization")
     model = AutoModelForSeq2SeqLM.from_pretrained("Shariar00/medical_summarization_finetune_medical_qa")
     text_generation_pipeline = pipeline("summarization", model=model, tokenizer=tokenizer)