ingsimo commited on
Commit
3a269fc
·
verified ·
1 Parent(s): 430d97b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -1,14 +1,14 @@
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', 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",
14
  model=model,
 
1
  from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
2
  import os
3
 
4
+ access_token=os.getenv('HF_TOKEN')
5
+ print(os.getenv('HF_TOKEN'))
6
 
7
  model_id = "iGeniusAI/Italia-9B-Instruct-v0.1"
8
+ '''
9
  model = AutoModelForCausalLM.from_pretrained(model_id,token='HF_TOKEN', 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",
14
  model=model,