brandyguillory commited on
Commit
a29d142
·
verified ·
1 Parent(s): 7a5c068

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -29,8 +29,8 @@ def load_model(model_name):
29
 
30
  return pipeline("text-generation", model=model_name, framework="tf")
31
 
32
- with warnings.catch_warnings():
33
- warnings.simplefilter("ignore")
34
  access_token = os.getenv("HF_ACCESS_TOKEN") # Use an environment variable for the access token
35
  return pipeline("text-generation", model=model_name, framework="tf", use_auth_token=access_token) # Use TensorFlow framework
36
 
 
29
 
30
  return pipeline("text-generation", model=model_name, framework="tf")
31
 
32
+ with warnings.catch_warnings():
33
+ warnings.simplefilter("ignore")
34
  access_token = os.getenv("HF_ACCESS_TOKEN") # Use an environment variable for the access token
35
  return pipeline("text-generation", model=model_name, framework="tf", use_auth_token=access_token) # Use TensorFlow framework
36