ingsimo commited on
Commit
e585462
·
verified ·
1 Parent(s): a9b85ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -1,13 +1,13 @@
1
  from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
2
  import os
3
 
4
- access_token=os.getenv('MODEL_REPO_ID')
5
- print(os.getenv('MODEL_REPO_ID'))
6
 
7
  model_id = "iGeniusAI/Italia-9B-Instruct-v0.1"
8
 
9
- model = AutoModelForCausalLM.from_pretrained(model_id,token=access_token, trust_remote_code=True)
10
- tokenizer = AutoTokenizer.from_pretrained(model_id,token=access_token, trust_remote_code=True)
11
  '''
12
  t_pipeline = pipeline(
13
  "text-generation",
 
1
  from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
2
  import os
3
 
4
+ #access_token=os.getenv('MODEL_REPO_ID')
5
+ #print(os.getenv('MODEL_REPO_ID'))
6
 
7
  model_id = "iGeniusAI/Italia-9B-Instruct-v0.1"
8
 
9
+ model = AutoModelForCausalLM.from_pretrained(model_id,token='HF_TOKEN'n, trust_remote_code=True)
10
+ tokenizer = AutoTokenizer.from_pretrained(model_id,token='HF_TOKEN', trust_remote_code=True)
11
  '''
12
  t_pipeline = pipeline(
13
  "text-generation",