ingsimo commited on
Commit
6c323bf
·
verified ·
1 Parent(s): cf70e92

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,12 +3,12 @@ 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=True, trust_remote_code=True)
10
  tokenizer = AutoTokenizer.from_pretrained(model_id,token=True, trust_remote_code=True)
11
-
12
  t_pipeline = pipeline(
13
  "text-generation",
14
  model=model,
 
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=True, trust_remote_code=True)
10
  tokenizer = AutoTokenizer.from_pretrained(model_id,token=True, trust_remote_code=True)
11
+ '''
12
  t_pipeline = pipeline(
13
  "text-generation",
14
  model=model,