Ikeofai commited on
Commit
8aa1219
1 Parent(s): 0607ab6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ from transformers import AutoTokenizer, pipeline
11
 
12
  # Assuming you have it stored securely
13
 
14
- model_name="Ikeofai/outputs"
15
  max_length=200
16
  tokenizer = AutoTokenizer.from_pretrained(model_name)
17
  config = PeftConfig.from_pretrained(model_name,token=token)
 
11
 
12
  # Assuming you have it stored securely
13
 
14
+ model_name="Ikeofai/gemma-2b-for-python-v2"
15
  max_length=200
16
  tokenizer = AutoTokenizer.from_pretrained(model_name)
17
  config = PeftConfig.from_pretrained(model_name,token=token)