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