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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -3,6 +3,7 @@ import os
3
 
4
  access_token=os.getenv('MODEL_REPO_ID')
5
  print(os.getenv('MODEL_REPO_ID'))
 
6
  model_id = "iGeniusAI/Italia-9B-Instruct-v0.1"
7
 
8
  model = AutoModelForCausalLM.from_pretrained(model_id,token=True, trust_remote_code=True)
@@ -40,3 +41,4 @@ outputs = t_pipeline(
40
  num_return_sequences=1,
41
  )
42
  print(outputs[0]["generated_text"])
 
 
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)
 
41
  num_return_sequences=1,
42
  )
43
  print(outputs[0]["generated_text"])
44
+ '''