hunthinn commited on
Commit
6a95174
1 Parent(s): 0975142

add access token

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -1,6 +1,13 @@
1
-
2
  from flask import Flask, jsonify, request
3
  from transformers import GPT2Tokenizer, GPT2LMHeadModel
 
 
 
 
 
 
 
 
4
 
5
  # Load the fine-tuned model and tokenizer
6
  tokenizer_path = "gpt2"
 
 
1
  from flask import Flask, jsonify, request
2
  from transformers import GPT2Tokenizer, GPT2LMHeadModel
3
+ from huggingface_hub import HfFolder
4
+ import os
5
+
6
+ access_token = os.getenv("HF_ACCESS_TOKEN")
7
+
8
+ # Authenticate with Hugging Face
9
+
10
+ HfFolder.save_token(access_token)
11
 
12
  # Load the fine-tuned model and tokenizer
13
  tokenizer_path = "gpt2"