Alaaeldin commited on
Commit
484975f
·
verified ·
1 Parent(s): f52f1d6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ MODEL_NAME = "Alaaeldin/pubmedBERT-demo"
10
 
11
  @st.cache_resource
12
  def load_pipeline():
13
- # Use the pipeline for question-answering
14
  return pipeline("text-generation", model=MODEL_NAME, tokenizer=MODEL_NAME, use_auth_token=access_token)
15
 
16
  # Load the pipeline
 
10
 
11
  @st.cache_resource
12
  def load_pipeline():
13
+ # Load the model and tokenizer with authentication
14
  return pipeline("text-generation", model=MODEL_NAME, tokenizer=MODEL_NAME, use_auth_token=access_token)
15
 
16
  # Load the pipeline