brandyguillory commited on
Commit
d0ca1db
·
verified ·
1 Parent(s): ce6e64e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ def load_model(model_name):
37
  with warnings.catch_warnings():
38
  warnings.simplefilter("ignore")
39
  access_token = os.getenv("HF_ACCESS_TOKEN") # Use an environment variable for the access token
40
- return pipeline("text-generation", model=model_name, framework="tf", use_auth_token=access_token) # Use TensorFlow framework
41
 
42
 
43
  # Button to run the model
 
37
  with warnings.catch_warnings():
38
  warnings.simplefilter("ignore")
39
  access_token = os.getenv("HF_ACCESS_TOKEN") # Use an environment variable for the access token
40
+ return pipeline("text-generation", model=model_name, framework="tf", token=access_token) # Use TensorFlow framework
41
 
42
 
43
  # Button to run the model