abenkbp commited on
Commit
4abb3bf
·
1 Parent(s): d9e0329
Files changed (1) hide show
  1. data/models/llama3-1-70b.py +1 -2
data/models/llama3-1-70b.py CHANGED
@@ -9,8 +9,7 @@ app = Flask(__name__)
9
 
10
  # Load the API key and initialize the InferenceClient
11
  api_key = os.getenv("ai")
12
- login(api_key)
13
- client = InferenceClient("meta-llama/Meta-Llama-3.1-70B-Instruct")
14
 
15
  @app.route('/chat', methods=['POST'])
16
  @spaces.GPU()
 
9
 
10
  # Load the API key and initialize the InferenceClient
11
  api_key = os.getenv("ai")
12
+ client = InferenceClient("meta-llama/Meta-Llama-3.1-70B-Instruct",token=api_key)
 
13
 
14
  @app.route('/chat', methods=['POST'])
15
  @spaces.GPU()